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
15e53f3c
Commit
15e53f3c
authored
Oct 13, 2018
by
Toby Boyd
Browse files
typos and method rename
parent
26301e74
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
official/resnet/resnet_run_loop.py
official/resnet/resnet_run_loop.py
+3
-3
No files found.
official/resnet/resnet_run_loop.py
View file @
15e53f3c
...
...
@@ -180,7 +180,7 @@ def image_bytes_serving_input_fn(image_shape, dtype=tf.float32):
images
,
{
'image_bytes'
:
image_bytes_list
})
def
override_flags_and_set_envars_for_gpu_
private_
thread
s
(
flags_obj
):
def
override_flags_and_set_envars_for_gpu_thread
_pool
(
flags_obj
):
"""Override flags and set env_vars for performance.
These settings exist to test the difference between using stock settings
...
...
@@ -457,9 +457,9 @@ def resnet_main(
model_helpers
.
apply_clean
(
flags
.
FLAGS
)
# Ensures fl
o
g override logic is only executed if explicitly triggered
# Ensures fl
a
g override logic is only executed if explicitly triggered
.
if
flags_obj
.
tf_gpu_thread_mode
:
override_flags_and_set_envars_for_gpu_
private_
thread
s
(
flags_obj
)
override_flags_and_set_envars_for_gpu_thread
_pool
(
flags_obj
)
# Create session config based on values of inter_op_parallelism_threads and
# intra_op_parallelism_threads. Note that we default to having
...
...
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