ne funcione pas

tutorials/image/cifar10/cifar10_train.py

@@ -36,6 +36,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
+import argparse
 from datetime import datetime
 import time
 
@@ -43,17 +44,15 @@ import tensorflow as tf
 
 import cifar10
 
-FLAGS = tf.app.flags.FLAGS
+parser = argparse.ArgumentParser()
 
-tf.app.flags.DEFINE_string('train_dir', '/tmp/cifar10_train',
-                           """Directory where to write event logs """
-                           """and checkpoint.""")
-tf.app.flags.DEFINE_integer('max_steps', 1000000,
-                            """Number of batches to run.""")
-tf.app.flags.DEFINE_boolean('log_device_placement', False,
-                            """Whether to log device placement.""")
-tf.app.flags.DEFINE_integer('log_frequency', 10,
-                            """How often to log results to the console.""")
+parser.add_argument('--train_dir', type=str, default='/tmp/cifar10_train', help='Directory where to write event logs and checkpoint.')
+
+parser.add_argument('--max_steps', type=int, default=1000000, help='Number of batches to run.')
+
+parser.add_argument('--log_device_placement', type=bool, default=False, help='Whether to log device placement.')
+
+parser.add_argument('--log_frequency', type=int, default=10, help='How often to log results to the console.')
 
 
 def train():
@@ -124,4 +123,5 @@ def main(argv=None):  # pylint: disable=unused-argument
 
 
 if __name__ == '__main__':
+  FLAGS = parser.parse_args()
   tf.app.run()