diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-04-20 03:04:48 -0700 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-04-20 03:04:48 -0700 |
| commit | 5b177789285aced41c690d5f0d4b412e17693c22 (patch) | |
| tree | 62cbfbd20bedc705df7dc319466622f467553127 /models/pix2pix_model.py | |
| parent | 3aeb748074628fd033d0fa798813aa226b8645d6 (diff) | |
| parent | 91e3983d2edd25f61366026e27d0ae2ae9d61282 (diff) | |
Merge branch 'master' of https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
Diffstat (limited to 'models/pix2pix_model.py')
| -rw-r--r-- | models/pix2pix_model.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/models/pix2pix_model.py b/models/pix2pix_model.py index 1d89b29..3bdd237 100644 --- a/models/pix2pix_model.py +++ b/models/pix2pix_model.py @@ -8,7 +8,7 @@ import util.util as util from util.image_pool import ImagePool from .base_model import BaseModel from . import networks - +from pdb import set_trace as st class Pix2PixModel(BaseModel): def name(self): @@ -108,6 +108,7 @@ class Pix2PixModel(BaseModel): self.loss_G.backward() def optimize_parameters(self): + # st() self.forward() self.optimizer_D.zero_grad() @@ -132,9 +133,8 @@ class Pix2PixModel(BaseModel): return OrderedDict([('real_A', real_A), ('fake_B', fake_B), ('real_B', real_B)]) def save(self, label): - use_gpu = self.gpu_ids is not None - self.save_network(self.netG, 'G', label, use_gpu) - self.save_network(self.netD, 'D', label, use_gpu) + self.save_network(self.netG, 'G', label, self.gpu_ids) + self.save_network(self.netD, 'D', label, self.gpu_ids) def update_learning_rate(self): lrd = self.opt.lr / self.opt.niter_decay |