Commit 560b3af4 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.

--synthetic_data, --dtype, --all_reduce_alg, and --num_packs have been unexposed from models which do not use them

PiperOrigin-RevId: 265483564
parent 0b3a8abf
...@@ -90,7 +90,8 @@ def create_model(data_format): ...@@ -90,7 +90,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(inter_op=True, intra_op=True, flags_core.define_performance(inter_op=True, intra_op=True,
num_parallel_calls=False) num_parallel_calls=False,
all_reduce_alg=True)
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',
......
...@@ -727,6 +727,10 @@ def define_resnet_flags(resnet_size_choices=None, dynamic_loss_scale=False, ...@@ -727,6 +727,10 @@ def define_resnet_flags(resnet_size_choices=None, dynamic_loss_scale=False,
flags_core.define_performance(num_parallel_calls=False, flags_core.define_performance(num_parallel_calls=False,
inter_op=True, inter_op=True,
intra_op=True, intra_op=True,
synthetic_data=True,
dtype=True,
all_reduce_alg=True,
num_packs=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,
......
...@@ -54,8 +54,8 @@ def get_loss_scale(flags_obj, default_for_fp16): ...@@ -54,8 +54,8 @@ def get_loss_scale(flags_obj, default_for_fp16):
def define_performance(num_parallel_calls=False, inter_op=False, intra_op=False, def define_performance(num_parallel_calls=False, inter_op=False, intra_op=False,
synthetic_data=True, max_train_steps=False, dtype=True, synthetic_data=False, max_train_steps=False, dtype=False,
all_reduce_alg=True, num_packs=True, all_reduce_alg=False, num_packs=False,
tf_gpu_thread_mode=False, tf_gpu_thread_mode=False,
datasets_num_private_threads=False, datasets_num_private_threads=False,
datasets_num_parallel_batches=False, datasets_num_parallel_batches=False,
......
...@@ -25,7 +25,8 @@ def define_flags(): ...@@ -25,7 +25,8 @@ def define_flags():
flags_core.define_base(num_gpu=False) flags_core.define_base(num_gpu=False)
flags_core.define_performance( flags_core.define_performance(
num_parallel_calls=True, inter_op=True, intra_op=True, num_parallel_calls=True, inter_op=True, intra_op=True,
dynamic_loss_scale=True, loss_scale=True) dynamic_loss_scale=True, loss_scale=True, synthetic_data=True,
dtype=True)
flags_core.define_image() flags_core.define_image()
flags_core.define_benchmark() flags_core.define_benchmark()
......
...@@ -249,6 +249,10 @@ def define_keras_flags(dynamic_loss_scale=True): ...@@ -249,6 +249,10 @@ def define_keras_flags(dynamic_loss_scale=True):
"""Define flags for Keras models.""" """Define flags for Keras models."""
flags_core.define_base(run_eagerly=True) flags_core.define_base(run_eagerly=True)
flags_core.define_performance(num_parallel_calls=False, flags_core.define_performance(num_parallel_calls=False,
synthetic_data=True,
dtype=True,
all_reduce_alg=True,
num_packs=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,
......
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