1. 程式人生 > >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()))