import tensorflow as tf

flags.DEFINE_integer("num_gpus", 2, "Total number of GPUs to use.")
flags.DEFINE_bool("multi_worker", True, "Multi-worker training.")

# My additional flags
tf.app.flags.DEFINE_boolean("use_original_ckpt", True, "use original ckpt")
flags.DEFINE_integer("task_index", 0, "task_index")
flags.DEFINE_string(
    "worker", "localhost:3000,localhost:3001", "specify workers in the cluster"
