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
13b89b93
Commit
13b89b93
authored
Apr 16, 2018
by
pkulzc
Browse files
Sync to master.
parents
a4b6765a
cac90a0e
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
2 deletions
+5
-2
research/slim/train_image_classifier.py
research/slim/train_image_classifier.py
+4
-1
samples/core/get_started/eager.ipynb
samples/core/get_started/eager.ipynb
+1
-1
No files found.
research/slim/train_image_classifier.py
View file @
13b89b93
...
...
@@ -552,6 +552,8 @@ def main(_):
# Merge all summaries together.
summary_op
=
tf
.
summary
.
merge
(
list
(
summaries
),
name
=
'summary_op'
)
# Add config to avoid 'could not satisfy explicit device' problem
sess_config
=
tf
.
ConfigProto
(
allow_soft_placement
=
True
)
###########################
# Kicks off the training. #
...
...
@@ -567,7 +569,8 @@ def main(_):
log_every_n_steps
=
FLAGS
.
log_every_n_steps
,
save_summaries_secs
=
FLAGS
.
save_summaries_secs
,
save_interval_secs
=
FLAGS
.
save_interval_secs
,
sync_optimizer
=
optimizer
if
FLAGS
.
sync_replicas
else
None
)
sync_optimizer
=
optimizer
if
FLAGS
.
sync_replicas
else
None
,
session_config
=
sess_config
)
if
__name__
==
'__main__'
:
...
...
samples/core/get_started/eager.ipynb
View file @
13b89b93
...
...
@@ -538,7 +538,7 @@
"\n",
"<table>\n",
" <tr><td>\n",
" <img src=\"http://cs231n.github.io/assets/nn3/opt1.gif\" width=\"70%\"\n",
" <img src=\"http
s
://cs231n.github.io/assets/nn3/opt1.gif\" width=\"70%\"\n",
" alt=\"Optimization algorthims visualized over time in 3D space.\">\n",
" </td></tr>\n",
" <tr><td align=\"center\">\n",
...
...
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