Unverified Commit 69c75593 authored by Jon Shlens's avatar Jon Shlens Committed by GitHub
Browse files

Merge pull request #3581 from chengpohi/patch-1

replace  `FLAGS.batch_size` by `images.get_shape()[0]`
parents 36abfad3 4a733c29
...@@ -240,7 +240,7 @@ def inference(images): ...@@ -240,7 +240,7 @@ def inference(images):
# local3 # local3
with tf.variable_scope('local3') as scope: with tf.variable_scope('local3') as scope:
# Move everything into depth so we can perform a single matrix multiply. # Move everything into depth so we can perform a single matrix multiply.
reshape = tf.reshape(pool2, [FLAGS.batch_size, -1]) reshape = tf.reshape(pool2, [images.get_shape()[0], -1])
dim = reshape.get_shape()[1].value dim = reshape.get_shape()[1].value
weights = _variable_with_weight_decay('weights', shape=[dim, 384], weights = _variable_with_weight_decay('weights', shape=[dim, 384],
stddev=0.04, wd=0.004) 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