Solved by collinarnett in post #3 
              
              
                My solution ended up being: 
from architectures import (Discriminator16, Discriminator64, Discriminator128, Generator16, Generator64, Generator128)

 # Create the Discriminator
 if ndf == 16:
   netD = Discriminator16(ngpu).to(device)
 elif ndf == 64:
   netD = Discriminator64(ngpu).to(device)
 elif…