How to check the total amount of network parameters in pytorch?

This article is about how to check the total network parameters in pytorch. Xiaobian thinks it is quite practical, so share it with everyone for reference. Let's follow Xiaobian and have a look.

netG = Generator()print ('#generator parameters:', sum(parameters am.numel() for parameters am in netG.parameters())netD = Discriminator()print ('#discriminator parameters:', sum(parameters am.numel() for parameters am in netD.parameters())

Add: PyTorch View parameters of network model PARAMS and FLOPS etc.

In PyTorch, you can use torchstat to view some information about the network model, including the total number of parameters params, MAdd, graphics memory footprint, FLOPs, etc.

The sample code is as follows: from torchstat import from torchvision.models import resnet50, resnet101, resnet152, resnext101_32x8dmodel = resnet50()stat(model, (3, 224, 224)) Thank you for reading! About "how to view the total number of network parameters in pytorch" this article is shared here, I hope the above content can be of some help to everyone, so that everyone can learn more knowledge, if you think the article is good, you can share it to let more people see it!