def get_args()

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()