Skip to content

Commit 0120dfe

Browse files
authored
Merge pull request #3810 from aaaasssddf/master
fixed dimension compatibility issue for numpy
2 parents a47a9d9 + e934a4a commit 0120dfe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tutorials/image/cifar10/cifar10.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,7 @@ def inference(images):
240240
# local3
241241
with tf.variable_scope('local3') as scope:
242242
# Move everything into depth so we can perform a single matrix multiply.
243-
reshape = tf.reshape(pool2, [images.get_shape()[0], -1])
243+
reshape = tf.reshape(pool2, [images.get_shape().as_list()[0], -1])
244244
dim = reshape.get_shape()[1].value
245245
weights = _variable_with_weight_decay('weights', shape=[dim, 384],
246246
stddev=0.04, wd=0.004)

0 commit comments

Comments
 (0)