Skip to content

GitLab

Unverified Commit 79264e3e authored by Martin Wicke's avatar Martin Wicke Committed by GitHub
Browse files

Merge pull request #2776 from alexgorban/master 

Attention OCR: Add missing is_training to batch_norm and dropout
parents d07447a3 e1ac0154
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
research/attention_ocr/python/model.py
View file @ 79264e3e
...@@ -201,9 +201,11 @@ class Model(object): ...@@ -201,9 +201,11 @@ class Model(object):
with tf.variable_scope('conv_tower_fn/INCE'): with tf.variable_scope('conv_tower_fn/INCE'):
if reuse: if reuse:
tf.get_variable_scope().reuse_variables() tf.get_variable_scope().reuse_variables()
with slim.arg_scope(inception.inception_v3_arg_scope()): with slim.arg_scope(
net, _ = inception.inception_v3_base( [slim.batch_norm, slim.dropout], is_training=is_training):
images, final_endpoint=mparams.final_endpoint) with slim.arg_scope(inception.inception_v3_arg_scope()):
net, _ = inception.inception_v3_base(
images, final_endpoint=mparams.final_endpoint)
return net return net
def _create_lstm_inputs(self, net): def _create_lstm_inputs(self, net):
......
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