diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2016-10-27 08:43:57 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-10-27 09:49:22 -0700 |
| commit | 6812d46b957e32eba37c67384cc2136908d7a1ff (patch) | |
| tree | 49e928e882aa2e44d9a8d627d0a35ae58683f92b /tensorflow/examples/image_retraining | |
| parent | ef7ac603d0dab2bb586a24dbbf0c6c669e213ccd (diff) | |
Parse argparse flags, then pass unparsed flags through to argv via tf.app.run()
to allow argparse + tf.flags to play nicely with each other.
Change: 137405161
Diffstat (limited to 'tensorflow/examples/image_retraining')
| -rw-r--r-- | tensorflow/examples/image_retraining/retrain.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/examples/image_retraining/retrain.py b/tensorflow/examples/image_retraining/retrain.py index 4f06cb8add..392f0176d3 100644 --- a/tensorflow/examples/image_retraining/retrain.py +++ b/tensorflow/examples/image_retraining/retrain.py @@ -1009,6 +1009,5 @@ if __name__ == '__main__': input pixels up or down by.\ """ ) - FLAGS = parser.parse_args() - - tf.app.run() + FLAGS, unparsed = parser.parse_known_args() + tf.app.run(main=main, argv=[sys.argv[0]] + unparsed) |