pytorch prints the total size of the parameters

#model 为你要打印的参数模型
print("Total number of paramerters in networks is {}  ".format(sum(x.numel() for x in model.parameters())))

Guess you like

Origin blog.csdn.net/qq_43360777/article/details/106382789