def __init__(self, opt):
super(ModelPlain4, self).__init__(opt)
# ------------------------------------
# define network
# ------------------------------------
self.netG = define_G(opt).to(self.device)
self.netG = DataParallel(self.netG)
def __init__(self, opt):
super(ModelPlain2, self).__init__(opt)
# ------------------------------------
# define network
# ------------------------------------
self.netG = define_G(opt)
self.netG = self.model_to_device(self.netG)
def __init__(self, opt):
super(ModelGAN, self).__init__(opt)
# ------------------------------------
# define network
# ------------------------------------
self.netG = define_G(opt)
self.netG = self.model_to_device(self.netG)
if self.is_train:
self.netD = define_D(opt)
self.netD = self.model_to_device(self.netD)
def __init__(self, opt):
super(ModelGAN, self).__init__(opt)
# ------------------------------------
# define network
# ------------------------------------
self.netG = define_G(opt).to(self.device)
self.netG = DataParallel(self.netG)
if self.is_train:
self.netF = define_F(opt).to(self.device)
self.netD = define_D(opt).to(self.device)
self.netF = DataParallel(self.netF)
self.netD = DataParallel(self.netD)