from ultralytics import YOLO
# 加载状态字典
#model = torch.load("/home/lab/LJ/wampee/ultralytics/yolov8n.pt")
model = YOLO('yolov8n.pt')  # 加载预训练模型
state_dict = model.state_dict()
lens=0
for name, param in state_dict.items():
    print(f"Parameter: {name}, Shape: {param.shape}")