Fix output channels for images with n_channels!=3

daa233 · daa233 · commit f002adfb7344 · 2019-07-28T17:56:56.000+08:00
diff --git a/configs/config.yaml b/configs/config.yaml
@@ -41,7 +41,7 @@ wgan_gp_lambda: 10
 
 # network parameters
 netG:
-  input_dim: 5
+  input_dim: 3
   ngf: 32
 
 netD:
