diff options
Diffstat (limited to 'models/test_model.py')
| -rw-r--r-- | models/test_model.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/models/test_model.py b/models/test_model.py index 03aef65..4af1fe1 100644 --- a/models/test_model.py +++ b/models/test_model.py @@ -17,6 +17,7 @@ class TestModel(BaseModel): self.netG = networks.define_G(opt.input_nc, opt.output_nc, opt.ngf, opt.which_model_netG, opt.norm, not opt.no_dropout, + opt.init_type, self.gpu_ids) which_epoch = opt.which_epoch self.load_network(self.netG, 'G', which_epoch) |