0
点赞
收藏
分享

微信扫一扫

pytorch统计模型的参数量


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



举报

相关推荐

0 条评论