pytorch------>>>统计模型参数量
total_num = sum(p.numel() for p in model.parameters())
trainable_num = sum(p.numel() for p in model.parameters() if p.requires_grad)
print('Total ',total_num)
print('Trainable ',trainable_num)
- 上一篇 »JavaScript文本框统计字数
- 下一篇 »python之计数统计