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:一行代码查看网络参数总量
最新推荐文章于 2024-08-08 08:26:21 发布