Handsome_zhou 发表于 2022-11-19 00:41:25

查看模型的参数量

本帖最后由 Handsome_zhou 于 2022-11-27 13:41 编辑

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

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

numel()函数用来返回列表或元组中元素个数。
页: [1]
查看完整版本: 查看模型的参数量