fix: 启用 TensorFloat32 加速矩阵乘法并解决 UserWarning

src/model/trainer.py

@@ -878,6 +878,10 @@ def train(
     """
     torch.multiprocessing.set_sharing_strategy("file_system")
 
+    # 启用 TensorFloat32 加速矩阵乘法 (解决 UserWarning 并提升性能)
+    if torch.cuda.is_available():
+        torch.set_float32_matmul_precision("high")
+
     # 设置随机种子
     torch.manual_seed(seed)
     if torch.cuda.is_available():