Python define_G Examples

Example #1

File: model_plain4.py Project: znsc/KAIR

 def __init__(self, opt):
     super(ModelPlain4, self).__init__(opt)
     # ------------------------------------
     # define network
     # ------------------------------------
     self.netG = define_G(opt).to(self.device)
     self.netG = DataParallel(self.netG)

Example #2

File: model_plain2.py Project: brstar96/KAIR

 def __init__(self, opt):
     super(ModelPlain2, self).__init__(opt)
     # ------------------------------------
     # define network
     # ------------------------------------
     self.netG = define_G(opt)
     self.netG = self.model_to_device(self.netG)

Example #3

 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)

Example #4

File: model_gan.py Project: zongking123/KAIR

 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)