print("Total number of paramerters in networks is {} ".format(sum(x.numel() for x in net.parameters())))
其中net是代码中的网络模型
print("Total number of paramerters in networks is {} ".format(sum(x.numel() for x in net.parameters())))
其中net是代码中的网络模型