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)