import torch import torchvision.models as models # 加载模型 model = models.resnet50() # 计算参数量 total_params = sum(p.numel() for p in model.parameters()) print(f"总参数量: {total_params}") # 计算可训练的参数量 trainable_params = sum(p.numel() for p in model.parameters() if p.requires_grad) print(f"可训练的参数量: {trainable_params}")