netG = Generator(ngpu).to(device)
# Handle multi-gpu if desired
if (device.type == 'cuda') and (ngpu > 1):
netG = nn.DataParallel(netG, list(range(ngpu)))
netG.apply(weights_init)
print(netG)
Created
February 13, 2020 06:44
-
-
Save SubhadityaMukherjee/350b46851d92393087dc461681d3a026 to your computer and use it in GitHub Desktop.
gen2
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment