Slide 26
Slide 26 text
© 2019, Amazon Web Services, Inc. or its affiliates. All rights reserved.
S U M M I T
S U M M I T © 2019, Amazon Web Services, Inc. or its affiliates. All rights reserved.
(Script Mode)
train.py
import argparse
if __name__ == '__main__’:
parser = argparse.ArgumentParser()
# hyperparameters
parser.add_argument('--epochs', type=int, default=10)
# input data and model directories
parser.add_argument('--train', type=str, default=os.environ['SM_CHANNEL_TRAIN'])
parser.add_argument('--test', type=str, default=os.environ['SM_CHANNEL_TEST'])
parser.add_argument('--model-dir', type=str, default=os.environ['SM_MODEL_DIR'])
args, _ = parser.parse_known_args()
/opt/ml/input/data/train
/opt/ml/input/data/test
/opt/ml/model