#model 为你要打印的参数模型
print("Total number of paramerters in networks is {} ".format(sum(x.numel() for x in model.parameters())))
pytorch prints the total size of the parameters
Guess you like
Origin blog.csdn.net/qq_43360777/article/details/106382789
Recommended
Ranking