Python save_networkの例

コード例 #1

 def save(self, epoch):
     util.save_network(self.netG, "G", epoch, self.opt)
     util.save_network(self.netD, "D", epoch, self.opt)
     if self.opt.use_vae:
         util.save_network(self.netE, "E", epoch, self.opt)
     if self.opt.use_blender:
         util.save_network(self.netB, "B", epoch, self.opt)
     if self.opt.unpairTrain:
         util.save_network(self.netD2, "D2", epoch, self.opt)

コード例 #2

 def save(self, epoch):
     util.save_network(self.netG, 'G', epoch, self.opt)
     util.save_network(self.netD, 'D', epoch, self.opt)
     if self.opt.use_vae:
         util.save_network(self.netE, 'E', epoch, self.opt)
     if self.opt.use_blender:
         util.save_network(self.netB, 'B', epoch, self.opt)
     if self.opt.unpairTrain:
         util.save_network(self.netD2, 'D2', epoch, self.opt)

コード例 #3

ファイル: pix2pix_model.py プロジェクト: Ugness/Line-Art-Colorization-SPADE

 def save(self, epoch):
     util.save_network(self.netG, 'G', epoch, self.opt)
     util.save_network(self.netD, 'D', epoch, self.opt)
     if self.opt.use_vae:
         util.save_network(self.netE, 'E', epoch, self.opt)
     if self.opt.use_F:
         util.save_network(self.netF, 'F', epoch, self.opt)

コード例 #4

 def save(self, epoch):
     util.save_network(self.net['netG'], 'G', epoch, self.opt)
     util.save_network(self.net['netD'], 'D', epoch, self.opt)
     util.save_network(self.net['netCorr'], 'Corr', epoch, self.opt)
     if self.opt.weight_domainC > 0 and self.opt.domain_rela:
         util.save_network(self.net['netDomainClassifier'],
                           'DomainClassifier', epoch, self.opt)

コード例 #5

 def save(self, epoch):
     util.save_network(self.netG, 'G', epoch, self.opt)
     util.save_network(self.netD, 'D', epoch, self.opt)
     if self.opt.train_rotate:
         util.save_network(self.netD_rotate, 'D_rotate', epoch, self.opt)
     if self.opt.use_vae:
         util.save_network(self.netE, 'E', epoch, self.opt)

コード例 #6

ファイル: pix2pix_trainer.py プロジェクト: praeclarumjj3/Experiments-CoCosNet

    def save(self, epoch):
        self.pix2pix_model_on_one_gpu.save(epoch)
        if self.opt.use_ema:
            self.netG_ema.assign(self.pix2pix_model_on_one_gpu.net['netG'])
            util.save_network(self.pix2pix_model_on_one_gpu.net['netG'],
                              'G_ema', epoch, self.opt)
            self.netG_ema.resume(self.pix2pix_model_on_one_gpu.net['netG'])

            self.netCorr_ema.assign(
                self.pix2pix_model_on_one_gpu.net['netCorr'])
            util.save_network(self.pix2pix_model_on_one_gpu.net['netCorr'],
                              'netCorr_ema', epoch, self.opt)
            self.netCorr_ema.resume(
                self.pix2pix_model_on_one_gpu.net['netCorr'])
        if epoch == 'latest':
            torch.save(
                {
                    'G': self.optimizer_G.state_dict(),
                    'D': self.optimizer_D.state_dict(),
                    'lr': self.old_lr,
                },
                os.path.join(self.opt.checkpoints_dir, self.opt.name,
                             'optimizer.pth'))

コード例 #7

ファイル: cifaredge_interp_model.py プロジェクト: arthur-qiu/SPADE

    def save(self, epoch):
        util.save_network(self.netG, 'G', epoch, self.opt)
        util.save_network(self.netD, 'D', epoch, self.opt)
        if self.opt.use_vae:
            util.save_network(self.netE, 'E', epoch, self.opt)

        if self.opt.cls:
            torch.save(
                self.net.state_dict(),
                os.path.join(
                    self.opt.save, self.opt.dataset + self.opt.cls_model +
                    '_epoch_' + str(epoch) + '.pt'))

コード例 #8

ファイル: model.py プロジェクト: nakamura-cv/SemIA

 def save(self, iter):
     util.save_network(self.netG, 'G', iter, self.opt)

コード例 #9

 def save(self, epoch):
     util.save_network(self.netG, 'G', epoch, self.opt)
     util.save_network(self.netD, 'D', epoch, self.opt)

コード例 #10

 def save(self, epoch):
     util.save_network(self.netG, "G", epoch, self.opt)
     util.save_network(self.netD, "D", epoch, self.opt)
     if self.opt.use_vae:
         util.save_network(self.netE, "E", epoch, self.opt)

コード例 #11

 def save(self, epoch):
     if self.opt.two_step_model:
         util.save_network(self.netGbg, 'Gbg', epoch, self.opt)
     util.save_network(self.netG, 'G', epoch, self.opt)
     util.save_network(self.netD, 'D', epoch, self.opt)

コード例 #12

ファイル: av_model.py プロジェクト: ycMia/Talking-Face_PC-AVS

 def save(self, epoch):
     if self.opt.train_recognition:
         util.save_network(self.netV, 'V', epoch, self.opt)
     elif self.opt.train_sync:
         util.save_network(self.netE, 'E', epoch, self.opt)
         if self.opt.use_audio:
             util.save_network(self.netA_sync, 'A_sync', epoch, self.opt)
     else:
         util.save_network(self.netG, 'G', epoch, self.opt)
         # util.save_network(self.netD, 'D', epoch, self.opt)
         if self.opt.use_audio:
             if self.opt.use_audio_id:
                 util.save_network(self.netA, 'A', epoch, self.opt)
             util.save_network(self.netA_sync, 'A_sync', epoch, self.opt)
         util.save_network(self.netV, 'V', epoch, self.opt)
         util.save_network(self.netE, 'E', epoch, self.opt)

コード例 #13

ファイル: m2c_model.py プロジェクト: Echo-wu151/SpineC2M-pytorch

 def save(self, epoch):
     util.save_network(self.netG_for_CT 'G_for_CT', epoch, self.opt)
     util.save_network(self.netD_aligned, 'D_aligned', epoch, self.opt)
     util.save_network(self.netG_for_MR, 'G_for_MR', epoch, self.opt)
     util.save_network(self.netD_unaligned, 'D_unaligned', epoch, self.opt)

コード例 #14

 def save(self, epoch):
     util.save_network(self.netG, 'G', epoch, self.opt)
     util.save_network(self.netD1, 'D1', epoch, self.opt)
     util.save_network(self.netD2, 'D2', epoch, self.opt)
     if self.opt.use_vae:
         util.save_network(self.netE, 'E', epoch, self.opt)

コード例 #15

 def save(self, epoch, triple=False):
     util.save_network(self.netG, 'G', epoch, self.opt, triple)
     util.save_network(self.netD, 'D', epoch, self.opt, triple)
     if self.opt.use_vae:
         util.save_network(self.netE, 'E', epoch, self.opt, triple)

コード例 #16

 def save(self, epoch):
     util.save_network(self.netSR, 'SR', epoch, self.opt)
     util.save_network(self.netD, 'D', epoch, self.opt)
     if self.use_E:
         util.save_network(self.netE, 'E', epoch, self.opt)

コード例 #17

 def save(self, epoch):
     util.save_network(self.netG, 'G', epoch, self.opt)
     util.save_network(self.netD, 'D', epoch, self.opt)
     util.save_network(self.netD_uncond, 'D_uncond', epoch, self.opt)
     if self.opt.use_vae:
         util.save_network(self.netE, 'E', epoch, self.opt)

コード例 #18

ファイル: OpenEdit_model.py プロジェクト: xh-liu/Open-Edit

 def save(self, epoch):
     util.save_network(self.netG, 'G', epoch, self.opt)
     if self.loss_gan:
         util.save_network(self.netD, 'D', epoch, self.opt)