#model 为你要打印的参数模型
print("Total number of paramerters in networks is {} ".format(sum(x.numel() for x in model.parameters())))
pytorch 打印参数的总规模
最新推荐文章于 2024-03-18 15:14:19 发布
#model 为你要打印的参数模型
print("Total number of paramerters in networks is {} ".format(sum(x.numel() for x in model.parameters())))