in benchmarks/tf_benchmarks/execute_tensorflow_training.py [0:0]
def get_args():
parser = argparse.ArgumentParser()
parser.add_argument(
"-t", "--instance-types", nargs="+", help="<Required> Set flag", required=True
)
parser.add_argument("-r", "--role", required=True)
parser.add_argument("-w", "--wait", action="store_true")
parser.add_argument("--region", default="us-west-2")
parser.add_argument("--py-versions", nargs="+", help="<Required> Set flag", default=["py3"])
parser.add_argument(
"--checkpoint-path",
default=os.path.join(default_bucket(), "benchmarks", "checkpoints"),
help="The S3 location where the model checkpoints and tensorboard events are saved after training",
)
return parser.parse_known_args()