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)
相关文章：
相似解决方案
热门标签
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)
相关文章：