Skip to content

GitLab

Commit 3a9e9edb authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Merge pull request #7462 from reedwm:flag_simp 

PiperOrigin-RevId: 263858817
parents 1f2cebfa a2a8ff82
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/resnet/ctl/ctl_imagenet_main.py
View file @ 3a9e9edb
...@@ -283,4 +283,6 @@ if __name__ == '__main__': ...@@ -283,4 +283,6 @@ if __name__ == '__main__':
logging.set_verbosity(logging.INFO) logging.set_verbosity(logging.INFO)
keras_common.define_keras_flags() keras_common.define_keras_flags()
ctl_common.define_ctl_flags() ctl_common.define_ctl_flags()
flags.adopt_module_key_flags(keras_common)
flags.adopt_module_key_flags(ctl_common)
absl_app.run(main) absl_app.run(main)
official/resnet/keras/keras_imagenet_main.py
View file @ 3a9e9edb
...@@ -268,6 +268,7 @@ def run(flags_obj): ...@@ -268,6 +268,7 @@ def run(flags_obj):
def define_imagenet_keras_flags(): def define_imagenet_keras_flags():
keras_common.define_keras_flags() keras_common.define_keras_flags()
flags_core.set_defaults(train_epochs=90) flags_core.set_defaults(train_epochs=90)
flags.adopt_module_key_flags(keras_common)
def main(_): def main(_):
......
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