From f548616df0deefad7f22bab264774125ed0f64e5 Mon Sep 17 00:00:00 2001
From: tian <11429339@qq.com>
Date: Wed, 10 Jun 2026 10:34:13 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E6=8A=91=E5=88=B6PyTorch=20torch.load?=
 =?UTF-8?q?=E7=9A=84FutureWarning=E5=99=AA=E9=9F=B3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/model_trainer.py | 2 ++
 1 file changed, 2 insertions(+)

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