"docs/vscode:/vscode.git/clone" did not exist on "b6f5ba9a809fcd2e5b2c440f538c1ccc965a9e59"
Commit e1ac0154 authored by Alexander Gorban's avatar Alexander Gorban
Browse files

Add the missing is_training to batch_norm and dropout.

parent d07447a3
......@@ -201,9 +201,11 @@ class Model(object):
with tf.variable_scope('conv_tower_fn/INCE'):
if reuse:
tf.get_variable_scope().reuse_variables()
with slim.arg_scope(inception.inception_v3_arg_scope()):
net, _ = inception.inception_v3_base(
images, final_endpoint=mparams.final_endpoint)
with slim.arg_scope(
[slim.batch_norm, slim.dropout], is_training=is_training):
with slim.arg_scope(inception.inception_v3_arg_scope()):
net, _ = inception.inception_v3_base(
images, final_endpoint=mparams.final_endpoint)
return 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