查看模型的参数量

Handsome_zhou

第一步: 实例化模型: net = your_model()

第二步: 计算并打印参数量: print('Total number of parameters in networks is {} '.format(sum(x.numel()  for x in net.parameters())))

numel()函数用来返回列表或元组中元素个数。