Slide 79
Slide 79 text
93
© 2020 Amazon Web Services, Inc. or its affiliates. All rights reserved |
コードの書き換え例: train.py (Script Mode, File Mode)
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
コマンド
ライン引数・
環境変数
から取得
Script Mode では普通の Python スクリプトとして実行される。
はじめに環境変数からデータ・モデル入出力のパスを取得して、
そこを読み書きするように train.py を書く。推論用にモデルを読み込む。