"...resnet50_tensorflow.git" did not exist on "4bb36073fd6e52a54f86bcc0c9a0e4a2efb46cb1"
Commit 15e53f3c authored by Toby Boyd's avatar Toby Boyd
Browse files

typos and method rename

parent 26301e74
...@@ -180,7 +180,7 @@ def image_bytes_serving_input_fn(image_shape, dtype=tf.float32): ...@@ -180,7 +180,7 @@ def image_bytes_serving_input_fn(image_shape, dtype=tf.float32):
images, {'image_bytes': image_bytes_list}) images, {'image_bytes': image_bytes_list})
def override_flags_and_set_envars_for_gpu_private_threads(flags_obj): def override_flags_and_set_envars_for_gpu_thread_pool(flags_obj):
"""Override flags and set env_vars for performance. """Override flags and set env_vars for performance.
These settings exist to test the difference between using stock settings These settings exist to test the difference between using stock settings
...@@ -457,9 +457,9 @@ def resnet_main( ...@@ -457,9 +457,9 @@ def resnet_main(
model_helpers.apply_clean(flags.FLAGS) model_helpers.apply_clean(flags.FLAGS)
# Ensures flog override logic is only executed if explicitly triggered # Ensures flag override logic is only executed if explicitly triggered.
if flags_obj.tf_gpu_thread_mode: if flags_obj.tf_gpu_thread_mode:
override_flags_and_set_envars_for_gpu_private_threads(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 # Create session config based on values of inter_op_parallelism_threads and
# intra_op_parallelism_threads. Note that we default to having # intra_op_parallelism_threads. Note that we default to having
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment