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
68b724ce
Unverified
Commit
68b724ce
authored
Feb 14, 2019
by
Toby Boyd
Committed by
GitHub
Feb 14, 2019
Browse files
Fix lint issues (#6204)
parent
5d3a7d04
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
0 deletions
+2
-0
official/resnet/estimator_cifar_benchmark.py
official/resnet/estimator_cifar_benchmark.py
+1
-0
official/resnet/keras/keras_imagenet_benchmark.py
official/resnet/keras/keras_imagenet_benchmark.py
+1
-0
No files found.
official/resnet/estimator_cifar_benchmark.py
View file @
68b724ce
...
@@ -106,6 +106,7 @@ class EstimatorCifar10BenchmarkTests(tf.test.Benchmark):
...
@@ -106,6 +106,7 @@ class EstimatorCifar10BenchmarkTests(tf.test.Benchmark):
self
.
_run_and_report_benchmark
()
self
.
_run_and_report_benchmark
()
def
_run_and_report_benchmark
(
self
):
def
_run_and_report_benchmark
(
self
):
"""Executes benchmark and reports result."""
start_time_sec
=
time
.
time
()
start_time_sec
=
time
.
time
()
stats
=
cifar_main
.
run_cifar
(
flags
.
FLAGS
)
stats
=
cifar_main
.
run_cifar
(
flags
.
FLAGS
)
wall_time_sec
=
time
.
time
()
-
start_time_sec
wall_time_sec
=
time
.
time
()
-
start_time_sec
...
...
official/resnet/keras/keras_imagenet_benchmark.py
View file @
68b724ce
...
@@ -67,6 +67,7 @@ class Resnet50KerasAccuracy(keras_benchmark.KerasBenchmark):
...
@@ -67,6 +67,7 @@ class Resnet50KerasAccuracy(keras_benchmark.KerasBenchmark):
self
.
_run_and_report_benchmark
()
self
.
_run_and_report_benchmark
()
def
benchmark_8_gpu_bfc_allocator
(
self
):
def
benchmark_8_gpu_bfc_allocator
(
self
):
"""Restricts CPU memory allocation."""
self
.
_setup
()
self
.
_setup
()
FLAGS
.
num_gpus
=
8
FLAGS
.
num_gpus
=
8
FLAGS
.
enable_eager
=
True
FLAGS
.
enable_eager
=
True
...
...
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