From ad7374438d59511bb340ce7f5ca720d34c897531 Mon Sep 17 00:00:00 2001 From: Taesung Park Date: Thu, 27 Apr 2017 01:32:10 -0700 Subject: changed the name of the loaded network from net_G_A to net_G --- models/one_direction_test_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'models/one_direction_test_model.py') 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) -- cgit v1.2.3-70-g09d2