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:一行代码查看网络参数总量
猜你喜欢
转载自blog.csdn.net/CV_YOU/article/details/85002754
今日推荐
周排行