////////////////////////////////////////////////////////
flags.DEFINE_string("dataset", "mnist", "The name of dataset [smallNORB, mnist, fashion-mnist]")
flags.DEFINE_boolean("is_training", True, "train or predict phase")
flags.DEFINE_integer("num_threads", 8, "number of threads of enqueueing exampls")
results = os.path.join("models", "results")
logdir = os.path.join(results, "logdir")
flags.DEFINE_string("results", results, "path for saving results")
flags.DEFINE_string("logdir", logdir, "logs directory")
After Change
// for training
flags.DEFINE_integer("batch_size", 128, "batch size")
flags.DEFINE_integer("num_steps", 50000, "The number of training steps, default: 50,000")
flags.DEFINE_integer("iter_routing", 3, "number of iterations in routing algorithm")
flags.DEFINE_integer("train_sum_every", 200, "the frequency of saving train summary(step)")
flags.DEFINE_integer("val_sum_every", 500, "the frequency of saving valuation summary(step)")
flags.DEFINE_integer("save_ckpt_every", 1000, "the frequency of saving model(step)")
