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
666d31b5
Commit
666d31b5
authored
Sep 21, 2017
by
Olivia
Browse files
twerk twerk twerk twerk twerk
parent
d34cf181
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
1 deletion
+5
-1
tutorials/image/cifar10/cifar10_train.py
tutorials/image/cifar10/cifar10_train.py
+5
-1
No files found.
tutorials/image/cifar10/cifar10_train.py
View file @
666d31b5
...
...
@@ -43,8 +43,9 @@ import time
import
tensorflow
as
tf
import
cifar10
from
cifar10
import
FLAGS
parser
=
argparse
.
ArgumentParser
()
parser
=
cifar10
.
parser
parser
.
add_argument
(
'--train_dir'
,
type
=
str
,
default
=
'/tmp/cifar10_train'
,
help
=
'Directory where to write event logs and checkpoint.'
)
...
...
@@ -95,6 +96,7 @@ def train():
self
.
_start_time
=
current_time
loss_value
=
run_values
.
results
# error on next line
examples_per_sec
=
FLAGS
.
log_frequency
*
FLAGS
.
batch_size
/
duration
sec_per_batch
=
float
(
duration
/
FLAGS
.
log_frequency
)
...
...
@@ -124,4 +126,6 @@ def main(argv=None): # pylint: disable=unused-argument
if
__name__
==
'__main__'
:
FLAGS
=
parser
.
parse_args
()
# for arg in vars(FLAGS):
# print(arg, getattr(FLAGS, arg))
tf
.
app
.
run
()
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