DeepSE / SageMakerActions

33 stars 16 forks source link

migration to sagemaker-python-sdk 2.x #68

Open JunYeopLee opened 4 years ago

JunYeopLee commented 4 years ago

It seems there are two minor changes in sagemaker-python-sdk 2.x.

If this tutorial is only for version 1, I think it would be great to specify the package version in requirements.txt

Estimators (2.x)

Renamed Estimator Parameters

https://github.com/aws/sagemaker-python-sdk/blob/64f600d677872fe8656cdf25d68fc4950b2cd28f/doc/v2.rst ... train_instance_count --> instance_count train_instance_type --> instance_type ...

Changes in Default Behavior

TensorFlow(
    entry_point="script.py",
    framework_version="2.2.0",  # now required
    py_version="py37",  # now required
    role="my-role",
    instance_type="ml.m5.xlarge",
    instance_count=1,
)
hunkim commented 3 years ago

Thanks!