diff --git a/sm_train.py b/sm_train.py index 111c897..bfe4f8c 100644 --- a/sm_train.py +++ b/sm_train.py @@ -5,7 +5,7 @@ from report import ResultReport sagemaker_session = sagemaker.Session(boto3.session.Session()) - + # Put the right role and input data role = "arn:aws:iam::294038372338:role/hunkimSagemaker" inputs = "s3://sagemaker-us-west-2-294038372338/sagemaker/hunkim-pytorch-mnist" @@ -34,7 +34,7 @@ }) estimator.fit({'training': inputs}) - + ######################################################################## # DONOT EDIT AFTER THIS LINE ########################################################################