diff options
Diffstat (limited to 'models/one_direction_test_model.py')
| -rw-r--r-- | models/one_direction_test_model.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/models/one_direction_test_model.py b/models/one_direction_test_model.py index 37e1bbb..d4f6442 100644 --- a/models/one_direction_test_model.py +++ b/models/one_direction_test_model.py @@ -22,9 +22,9 @@ class OneDirectionTestModel(BaseModel): opt.norm, opt.use_dropout, self.gpu_ids) which_epoch = opt.which_epoch - AtoB = self.opt.which_direction == 'AtoB' - which_network = 'G_A' if AtoB else 'G_B' - self.load_network(self.netG_A, which_network, which_epoch) + #AtoB = self.opt.which_direction == 'AtoB' + #which_network = 'G_A' if AtoB else 'G_B' + self.load_network(self.netG_A, 'G', which_epoch) print('---------- Networks initialized -------------') networks.print_network(self.netG_A) |