0785f0c037806584b000fe7b2b2ae8888980f3ed,tutorials/image/cifar10/cifar10_train.py,,,#,46

Before Change

import cifar10

parser = cifar10.parser

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():
  Train CIFAR-10 for a number of steps.

After Change

import cifar10

FLAGS = tf.app.flags.FLAGS

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.)

Instances