learning_rate = 1e-4 ####### model parameter ####### netG = VAE1().to(device); netD= Discriminator(nChannels=64,ndf=64).to(device); ####### optimizer ########### op