Commit 9fb1a1b6 authored by Soroush Radpour's avatar Soroush Radpour Committed by Toby Boyd
Browse files

Returning an object causes the program to exit with a non-zero code. (#7294)

parent c612d8c7
...@@ -269,7 +269,7 @@ def define_imagenet_keras_flags(): ...@@ -269,7 +269,7 @@ def define_imagenet_keras_flags():
def main(_): def main(_):
model_helpers.apply_clean(flags.FLAGS) model_helpers.apply_clean(flags.FLAGS)
with logger.benchmark_context(flags.FLAGS): with logger.benchmark_context(flags.FLAGS):
return run(flags.FLAGS) run(flags.FLAGS)
if __name__ == '__main__': if __name__ == '__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