fix: deconv_type assignation

307509256 · Apr 23, 2017 · d7fa1b9 · d7fa1b9
1 parent ea12768
commit d7fa1b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -68,7 +68,7 @@ def __init__(self, sess, args, devices, infer=False, name='SEGAN'):
         self.z_dim = args.z_dim
         self.z_depth = args.z_depth
         # type of deconv
-        self.deconv_type = deconv_type
+        self.deconv_type = args.deconv_type
         # specify if use biases or not
         self.bias_downconv = args.bias_downconv
         self.bias_deconv = args.bias_deconv