fix: 抑制PyTorch torch.load的FutureWarning噪音

2026-06-10 10:34:13 +08:00 · 2026-06-10 10:34:13 +08:00 · f548616df0
commit f548616df0
parent 79622678c7
1 changed files with 2 additions and 0 deletions
--- a/src/model_trainer.py
+++ b/src/model_trainer.py
@ -13,6 +13,8 @@ import math

 # PyTorch 为可选依赖
 try:
+    import warnings
+    warnings.filterwarnings('ignore', message='You are using `torch.load` with `weights_only=False`')
    import torch
    import torch.nn as nn
    from torch.utils.data import DataLoader