Slide 24
Slide 24 text
© 2020, Amazon Web Services, Inc. or its Affiliates.
import sagemaker
from sagemaker.mxnet import MXNet # 各フレームワークに対応した Estimator クラス
estimator = MXNet("train.py", # トレーニングスクリプトなどを指定して初期化
role=sagemaker.get_execution_role(),
train_instance_count=1,
train_instance_type="ml.p3.2xlarge",
framework_version="1.4.1")
estimator.fit("s3://mybucket/data/train") # fit でトレーニング
predictor = estimator.deploy(initial_instance_count=2, # 2以上にすると Multi-AZ
instance_type="ml.m5.xlarge") # deploy でエンドポイント作成
SageMaker Python SDK
環境をコードで表現して、それに合わせて実環境を動作させてくれる