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())