flags.DEFINE_string("output_base_path", "/tmp/swivel_data",
"Path where to write the trained embeddings.")
flags.DEFINE_integer("embedding_size", 300, "Size of the embeddings")
flags.DEFINE_boolean("trainable_bias", False, "Biases are trainable")
flags.DEFINE_integer("submatrix_rows", 4096, "Rows in each training submatrix. "
"This must match the training data.")
flags.DEFINE_integer("submatrix_cols", 4096, "Rows in each training submatrix. "
"This must match the training data.")
After Change
"Comma-separated list of parameter server host:port; if empty, run local")
flags.DEFINE_string(
"worker_hosts", "", "Comma-separated list of worker host:port")
flags.DEFINE_string(
"job_name", "", "The job this process will run, either "ps" or "worker"")
flags.DEFINE_integer(
"task_index", 0, "The task index for this process")
flags.DEFINE_integer(
"gpu_device", 0, "The GPU device to use.")
