33 lines
986 B
Python
33 lines
986 B
Python
#!/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✅ 所有导入测试通过！") |