From 7800d516596f1a25986b458cddf8b8785bcc7df8 Mon Sep 17 00:00:00 2001
From: junyanz <junyanz@berkeley.edu>
Date: Fri, 6 Oct 2017 10:46:43 -0700
Subject: support nc=1, add new leaerning rate policy and new initialization

---
 models/test_model.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'models/test_model.py')

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)
-- 
cgit v1.2.3-70-g09d2