pytorch中怎么查看网络参数总量呢

这篇文章给大家分享的是有关pytorch中怎么查看网络参数总量呢的内容。小编觉得挺实用的，因此分享给大家做个参考，一起跟随小编过来看看吧。

大家还是直接看代码吧~

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

补充：PyTorch查看网络模型的参数量PARAMS和FLOPS等

在PyTorch中，可以使用torchstat这个库来查看网络模型的一些信息，包括总的参数量params、MAdd、显卡内存占用量和FLOPs等。

示例代码如下：

from torchstat import statfrom torchvision.models import resnet50, resnet101, resnet152, resnext101_32x8dmodel = resnet50()stat(model, (3, 224, 224))

感谢各位的阅读！关于"pytorch中怎么查看网络参数总量呢"这篇文章就分享到这里了，希望以上内容可以对大家有一定的帮助，让大家可以学到更多知识，如果觉得文章不错，可以把它分享出去让更多的人看到吧！