torchsummary的使用
安装torchsummarypip install torchsummary例子netE = Encoder(NUM_CHANNELS=3)netG = Generator(NUM_CHANNELS=3, z_dim=1024, z_size=1)netD = Discriminator(NUM_CHANNELS=3, z_dim=1024)netD.to(device)netE.to(device)netG.to(device)summary(netE, input_size=(3,
原创
2020-12-02 09:45:24 ·
900 阅读 ·
0 评论