Commit a2a8ff82 authored by Reed's avatar Reed
Browse files

Show flags when --help is specified in resnet.

parent ff6c3b1e
......@@ -283,4 +283,6 @@ if __name__ == '__main__':
logging.set_verbosity(logging.INFO)
keras_common.define_keras_flags()
ctl_common.define_ctl_flags()
flags.adopt_module_key_flags(keras_common)
flags.adopt_module_key_flags(ctl_common)
absl_app.run(main)
......@@ -268,6 +268,7 @@ def run(flags_obj):
def define_imagenet_keras_flags():
keras_common.define_keras_flags()
flags_core.set_defaults(train_epochs=90)
flags.adopt_module_key_flags(keras_common)
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