netD = Discriminator(ngpu).to(device)
if (device.type == 'cuda') and (ngpu > 1):
netD = nn.DataParallel(netD, list(range(ngpu)))
netD.apply(weights_init)
print(netD)
Created
February 13, 2020 06:45
-
-
Save SubhadityaMukherjee/b3b066c1d3b8109daa47350680d88110 to your computer and use it in GitHub Desktop.
dis2
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment