0dc509d1f74b6aea3ec90c214e47ce26a1ec375f,swivel/swivel.py,,,#,66

Before Change

flags.DEFINE_integer("num_readers", 4,
                     "Number of threads to read the input data and feed it")
flags.DEFINE_float("num_epochs", 40, "Number epochs to train for")
flags.DEFINE_float("per_process_gpu_memory_fraction", 0,
                   "Fraction of GPU memory to use, 0 means allow_growth")
flags.DEFINE_integer("num_gpus", 0,
                     "Number of GPUs to use, 0 means all available")

FLAGS = flags.FLAGS

After Change

flags.DEFINE_string(
    "output_base_path", "/tmp/swivel_data",
    "Path where to write the trained embeddings.")
flags.DEFINE_string("eval_base_path", "", "Path to evaluation data")

// Control for training.
flags.DEFINE_float("num_epochs", 40, "Number epochs to train")
flags.DEFINE_string("hparams", "", "Model hyper-parameters")

Instances