diff options
| author | junyanz <junyanzhu89@gmail.com> | 2018-02-09 11:39:35 -0500 |
|---|---|---|
| committer | junyanz <junyanzhu89@gmail.com> | 2018-02-09 11:39:35 -0500 |
| commit | 0ae4f0500e415a6a67689ef9356e8e4779ae5833 (patch) | |
| tree | 25392e96a1b64c8454f7f548886af7dd48aa6bd0 /models/cycle_gan_model.py | |
| parent | 7a5e2cd5f5003e8ca9a0fc3dac14a74b81287881 (diff) | |
code reformatting
Diffstat (limited to 'models/cycle_gan_model.py')
| -rw-r--r-- | models/cycle_gan_model.py | 9 |
1 files changed, 1 insertions, 8 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index b7b840d..bcc6a15 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -1,6 +1,4 @@ -import numpy as np import torch -import os from collections import OrderedDict from torch.autograd import Variable import itertools @@ -8,7 +6,6 @@ import util.util as util from util.image_pool import ImagePool from .base_model import BaseModel from . import networks -import sys class CycleGANModel(BaseModel): @@ -17,10 +14,6 @@ class CycleGANModel(BaseModel): def initialize(self, opt): BaseModel.initialize(self, opt) - - nb = opt.batchSize - size = opt.fineSize - # load/define networks # The naming conversion is different from those used in the paper # Code (paper): G_A (G), G_B (F), D_A (D_Y), D_B (D_X) @@ -200,7 +193,7 @@ class CycleGANModel(BaseModel): def get_current_errors(self): ret_errors = OrderedDict([('D_A', self.loss_D_A), ('G_A', self.loss_G_A), ('Cyc_A', self.loss_cycle_A), - ('D_B', self.loss_D_B), ('G_B', self.loss_G_B), ('Cyc_B', self.loss_cycle_B)]) + ('D_B', self.loss_D_B), ('G_B', self.loss_G_B), ('Cyc_B', self.loss_cycle_B)]) if self.opt.identity > 0.0: ret_errors['idt_A'] = self.loss_idt_A ret_errors['idt_B'] = self.loss_idt_B |