diff --git a/src/model_trainer.py b/src/model_trainer.py index 12769ed..70b9cf9 100644 --- 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