Commit 882e51a4 authored by Reed Wanderman-Milne's avatar Reed Wanderman-Milne Committed by A. Unique TensorFlower
Browse files

Unexpose some flags from models which do not use them.

--num_parallel_calls, --inter_op_parallelism_threads, and --intra_op_parallelism_threads have been unexposed from models which do not use them

PiperOrigin-RevId: 264965788
parent 74543c03
...@@ -89,7 +89,8 @@ def create_model(data_format): ...@@ -89,7 +89,8 @@ def create_model(data_format):
def define_mnist_flags(): def define_mnist_flags():
flags_core.define_base() flags_core.define_base()
flags_core.define_performance(num_parallel_calls=False) flags_core.define_performance(inter_op=True, intra_op=True,
num_parallel_calls=False)
flags_core.define_image() flags_core.define_image()
flags.adopt_module_key_flags(flags_core) flags.adopt_module_key_flags(flags_core)
flags_core.set_defaults(data_dir='/tmp/mnist_data', flags_core.set_defaults(data_dir='/tmp/mnist_data',
......
...@@ -725,6 +725,8 @@ def define_resnet_flags(resnet_size_choices=None, dynamic_loss_scale=False, ...@@ -725,6 +725,8 @@ def define_resnet_flags(resnet_size_choices=None, dynamic_loss_scale=False,
"""Add flags and validators for ResNet.""" """Add flags and validators for ResNet."""
flags_core.define_base() flags_core.define_base()
flags_core.define_performance(num_parallel_calls=False, flags_core.define_performance(num_parallel_calls=False,
inter_op=True,
intra_op=True,
tf_gpu_thread_mode=True, tf_gpu_thread_mode=True,
datasets_num_private_threads=True, datasets_num_private_threads=True,
dynamic_loss_scale=dynamic_loss_scale, dynamic_loss_scale=dynamic_loss_scale,
......
...@@ -53,7 +53,7 @@ def get_loss_scale(flags_obj, default_for_fp16): ...@@ -53,7 +53,7 @@ def get_loss_scale(flags_obj, default_for_fp16):
return default_for_fp16 return default_for_fp16
def define_performance(num_parallel_calls=True, inter_op=True, intra_op=True, def define_performance(num_parallel_calls=False, inter_op=False, intra_op=False,
synthetic_data=True, max_train_steps=False, dtype=True, synthetic_data=True, max_train_steps=False, dtype=True,
all_reduce_alg=True, num_packs=True, all_reduce_alg=True, num_packs=True,
tf_gpu_thread_mode=False, tf_gpu_thread_mode=False,
......
...@@ -23,7 +23,9 @@ from official.utils.flags import core as flags_core # pylint: disable=g-bad-imp ...@@ -23,7 +23,9 @@ from official.utils.flags import core as flags_core # pylint: disable=g-bad-imp
def define_flags(): def define_flags():
flags_core.define_base(num_gpu=False) flags_core.define_base(num_gpu=False)
flags_core.define_performance(dynamic_loss_scale=True, loss_scale=True) flags_core.define_performance(
num_parallel_calls=True, inter_op=True, intra_op=True,
dynamic_loss_scale=True, loss_scale=True)
flags_core.define_image() flags_core.define_image()
flags_core.define_benchmark() flags_core.define_benchmark()
......
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