diff --git a/ops.py b/ops.py index 8975f58..d4b5257 100644 --- a/ops.py +++ b/ops.py @@ -82,7 +82,7 @@ def deconv2d(input_, output_shape, name="deconv2d", with_w=False): with tf.variable_scope(name): # filter : [height, width, output_channels, in_channels] - w = tf.get_variable('w', [k_h, k_h, output_shape[-1], input_.get_shape()[-1]], + w = tf.get_variable('w', [k_h, k_w, output_shape[-1], input_.get_shape()[-1]], initializer=tf.random_normal_initializer(stddev=stddev)) try: