Unverified Commit 0120dfe6 authored by Jon Shlens's avatar Jon Shlens Committed by GitHub
Browse files

Merge pull request #3810 from aaaasssddf/master

fixed dimension compatibility issue for numpy
parents a47a9d95 e934a4ad
......@@ -240,7 +240,7 @@ def inference(images):
# local3
with tf.variable_scope('local3') as scope:
# Move everything into depth so we can perform a single matrix multiply.
reshape = tf.reshape(pool2, [images.get_shape()[0], -1])
reshape = tf.reshape(pool2, [images.get_shape().as_list()[0], -1])
dim = reshape.get_shape()[1].value
weights = _variable_with_weight_decay('weights', shape=[dim, 384],
stddev=0.04, wd=0.004)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment