Python define_Gの例

コード例 #1

ファイル: model_plain4.py プロジェクト: 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)

コード例 #2

ファイル: model_plain2.py プロジェクト: 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)

コード例 #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)

コード例 #4

ファイル: model_gan.py プロジェクト: 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)