aws / aws/aws-step-functions-data-science-sdk-python
timestamp mismatch when using code_location
- 主要语言
- Python
- 星标
- 299
- 派生
- 84
- PR 合并指标
- 30 天内没有已合并 PR
描述
HI,
When code_location is used in estimator of TrainingStep(), the uploaded s3 path and sagemaker_submit_directory timestamp do not match(about 400 ms).
This will cause the execution to fail.
In SageMaker training job, timestamp matches even if code_location is used.
S3 uploaded path
s3://my-bucket/model/sagemaker-xgboost-2020-06-10-06-29-37-910/source/sourcedir.tar.gz
sagemaker_submit_directory
"s3://my-bucket/model/sagemaker-xgboost-2020-06-10-06-29-38-323/source/sourcedir.tar.gz"
```
# Open Source distributed script mode
from sagemaker.session import s3_input, Session
from sagemaker.xgboost.estimator import XGBoost
boto_session = boto3.Session(region_name=region)
session = Session(boto_session=boto_session)
output_path = 's3://{}/{}'.format(bucket_name, 'model')
xgb_script_mode_estimator = XGBoost(
entry_point='xgboost.py',
source_dir='source',
framework_version='0.90-2', # Note: framework_version is mandatory
hyperparameters=hyperparams,
role=role,
train_instance_count=1,
train_instance_type='ml.m5.2xlarge',
code_location=output_path, # ← Cause a mismatch
output_path=output_path
)
```
贡献指南
调研方向
从复现中展示的 TrainingStep estimator 和 code_location 处理开始。运行提供的 XGBoost 示例,并将上传到 S3 的路径与 sagemaker_submit_directory 的时间戳进行比较。当两个路径使用相同的时间戳,从而使训练执行成功时,即表示完成。
由索引模型根据 Issue 内容生成。
评估
- 技术栈
- aws, python
- 领域
- cloud, machine-learning
- Issue 类型
- 缺陷
- 难度
- 4/5
- 预计耗时
- 3-5 天
- 活跃度
- 停滞
- 描述清晰度
- 基本清楚
- 新手友好度
- 25/100