diff --git a/TensorFlow/contrib/cv/Transferring-GAN_ID1252_for_Tensorflow/tflib/ops/conv2d.py b/TensorFlow/contrib/cv/Transferring-GAN_ID1252_for_Tensorflow/tflib/ops/conv2d.py index 3cec271ab8b854bb6ab9574e719088cc972ec62c..23f1a6ab5d31929438d92e6c38407902051246de 100644 --- a/TensorFlow/contrib/cv/Transferring-GAN_ID1252_for_Tensorflow/tflib/ops/conv2d.py +++ b/TensorFlow/contrib/cv/Transferring-GAN_ID1252_for_Tensorflow/tflib/ops/conv2d.py @@ -131,15 +131,15 @@ def Conv2D(name, input_dim, output_dim, filter_size, inputs, he_init=True, mask_ if mask_type is not None: with tf.name_scope('filter_mask'): filters = filters * mask - + result = tf.transpose(result, perm=[0, 2, 3, 1]) result = tf.nn.conv2d( input=inputs, filter=filters, strides=[1, 1, stride, stride], padding='SAME', - data_format='NCHW' + data_format='NHWC' ) - + result = tf.transpose(result,perm=[0, 3, 1, 2]) if biases: _biases = lib.param( name+'.Biases',