初步可跑通，但loss计算有问题，不收敛

test_import.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python3
+"""
+测试 timm 导入是否正常工作
+"""
+import sys
+print("Python version:", sys.version)
+
+try:
+    from timm.layers import to_2tuple, DropPath, trunc_normal_
+    from timm.models import register_model
+    print("✓ 成功导入 timm.layers.to_2tuple")
+    print("✓ 成功导入 timm.layers.DropPath")
+    print("✓ 成功导入 timm.layers.trunc_normal_")
+    print("✓ 成功导入 timm.models.register_model")
+except ImportError as e:
+    print(f"✗ 导入失败: {e}")
+    sys.exit(1)
+
+try:
+    from models.swiftformer import SwiftFormer_XS
+    print("✓ 成功导入 SwiftFormer_XS")
+except ImportError as e:
+    print(f"✗ 导入 SwiftFormer_XS 失败: {e}")
+    sys.exit(1)
+
+try:
+    from models.swiftformer_temporal import SwiftFormerTemporal_XS
+    print("✓ 成功导入 SwiftFormerTemporal_XS")
+except ImportError as e:
+    print(f"✗ 导入 SwiftFormerTemporal_XS 失败: {e}")
+    sys.exit(1)
+
+print("\n✅ 所有导入测试通过！")