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
29c28d8f
Commit
29c28d8f
authored
Mar 15, 2017
by
Neal Wu
Committed by
GitHub
Mar 15, 2017
Browse files
Rename to log_frequency and a few minor changes
parent
3bc7f6c2
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
11 deletions
+10
-11
tutorials/image/cifar10/cifar10_train.py
tutorials/image/cifar10/cifar10_train.py
+10
-11
No files found.
tutorials/image/cifar10/cifar10_train.py
View file @
29c28d8f
...
...
@@ -52,9 +52,8 @@ tf.app.flags.DEFINE_integer('max_steps', 1000000,
"""Number of batches to run."""
)
tf
.
app
.
flags
.
DEFINE_boolean
(
'log_device_placement'
,
False
,
"""Whether to log device placement."""
)
tf
.
app
.
flags
.
DEFINE_integer
(
'log_steps_count'
,
10
,
"""Log process results per count."""
)
tf
.
app
.
flags
.
DEFINE_integer
(
'log_frequency'
,
10
,
"""How often to log results to the console."""
)
def
train
():
...
...
@@ -88,16 +87,16 @@ def train():
return
tf
.
train
.
SessionRunArgs
(
loss
)
# Asks for loss value.
def
after_run
(
self
,
run_context
,
run_values
):
log_steps
=
FLAGS
.
log_steps_count
if
self
.
_step
%
log_steps
==
0
:
duration
=
time
.
time
()
-
self
.
_start_time
self
.
_start_time
=
time
.
time
()
log_frequency
=
FLAGS
.
log_frequency
loss_value
=
run_values
.
results
if
self
.
_step
%
log_frequency
==
0
:
current_time
=
time
.
time
()
duration
=
current_time
-
self
.
_start_time
self
.
_start_time
=
current_time
num_examples_per_step
=
FLAGS
.
batch_size
examples_per_sec
=
num_examples_per_step
*
FLAGS
.
log_steps_count
/
duration
sec_per_batch
=
float
(
duration
/
log_
steps
)
loss_value
=
run_values
.
results
examples_per_sec
=
log_frequency
*
FLAGS
.
batch_size
/
duration
sec_per_batch
=
float
(
duration
/
log_
frequency
)
format_str
=
(
'%s: step %d, loss = %.2f (%.1f examples/sec; %.3f '
'sec/batch)'
)
...
...
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