69d221deb018beed51b2f5216ca34c61cca92858,models/official/squeezenet/squeezenet_main.py,,,#,34

Before Change

    "will attempt to automatically detect the GCE project from metadata.")

// Model specific paramenters
flags.DEFINE_string("data_dir", "", "Location of training files.")
flags.DEFINE_string("model_dir", "", "Where to store model checkpoints.")
flags.DEFINE_integer("save_checkpoints_secs", 3600,
                     "Interval between saving model checkpoints.")
flags.DEFINE_integer("num_shards", 8, "Number of TPU shards.")
flags.DEFINE_integer("batch_size", 1024, "Batch size for training and eval.")
flags.DEFINE_boolean("use_tpu", True, "If true, use TPU device.")

flags.DEFINE_string("optimizer", "momentum", "Optimizer: momentum|adam|rmsprop")
flags.DEFINE_float("momentum", 0.9, "Momentum parameter for SGD optimizer.")
flags.DEFINE_integer("num_epochs", 150,
                     "Number of epochs of the training set to process.")
flags.DEFINE_integer("num_evals", 10,

After Change

import squeezenet_model
from configs import squeezenet_config

common_tpu_flags.define_common_tpu_flags()
common_hparams_flags.define_common_hparams_flags()

flags.DEFINE_integer("num_examples_per_epoch", None,
                     "Number of examples to train per epoch.")

Instances