#model 为你要打印的参数模型
print("Total number of paramerters in networks is {} ".format(sum(x.numel() for x in model.parameters())))
pytorch 打印参数的总规模
猜你喜欢
转载自blog.csdn.net/qq_43360777/article/details/106382789
今日推荐
周排行