print("--------模型总的参数量---------")
print(sum(p.numel() for p in model.parameters())) # 打印模型参数量
print("--------模型训练的参数量---------")
print(sum(p.numel() for p in model.parameters() if p.requires_grad)) # 打印模型参数量
#打印模型名称与shape
for name,parameters in model.named_parameters():
print(name,':',parameters.size())