Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
e1ac0154
"docs/vscode:/vscode.git/clone" did not exist on "b6f5ba9a809fcd2e5b2c440f538c1ccc965a9e59"
Commit
e1ac0154
authored
Nov 10, 2017
by
Alexander Gorban
Browse files
Add the missing is_training to batch_norm and dropout.
parent
d07447a3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
3 deletions
+5
-3
research/attention_ocr/python/model.py
research/attention_ocr/python/model.py
+5
-3
No files found.
research/attention_ocr/python/model.py
View file @
e1ac0154
...
...
@@ -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
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment