To pass arguments to training code in Sagemaker by itself or via a Step Function, set up an argument parser in the Python script with keyword arguments.
Example#
Implement an argument parser in the Sagemaker entry point script. For example, in a Python script:
import argparse
if __name__ == "__main__":
parser = argparse.ArgumentParser
parser.add_argument("--learning-rate", type=int, default=1)
parser.add_argument("--batch-size", type=int, default=64)
parser.add_argument("--communicator", type=str)
parser.add_argument("--frequency", type=int, default=20)
args = parser.parse_args
...
Start a training job with hyperparameters.
{"HyperParameters": {"batch-size": 256, "learning-rate": 0.0001, "communicator": "pure_nccl"}}