Skip to content

GitLab

Commit a2a8ff82 authored by Reed's avatar Reed
Browse files

Show flags when --help is specified in resnet.

parent ff6c3b1e
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/resnet/ctl/ctl_imagenet_main.py
View file @ a2a8ff82
...@@ -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 @ a2a8ff82
...@@ -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