diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-04-27 01:32:10 -0700 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-04-27 01:32:10 -0700 |
| commit | ad7374438d59511bb340ce7f5ca720d34c897531 (patch) | |
| tree | c1772116e733e4ad16175eee5010f953fe3a03fc | |
| parent | e5b2fd6d36b4297c4314478e88820cd10943d192 (diff) | |
changed the name of the loaded network from net_G_A to net_G
| -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) |