aws / aws/sagemaker-python-sdk

ModelTrainer API cannot configure the right Session

オープン
#5,237 コメント 5 件 リアクション 0 件 担当者 0 名 GitHub で見る
component: training type: bug
主要言語
Python
スター
2.3k
フォーク
1.3k
平均マージ
1日 22時間
マージ済み PR(30日)
35

説明

**Describe the bug**
I'm trying to train a model using the `sagemaker.modules.train.ModelTrainer` API. However, it keeps trying to validate the SageMaker session using Pydantic, only never to accept any possible input. It spits out the Validation Error you see in the screenshot attached below.

**To reproduce**
1. Write a ModelTrainer compatible script
2. Write the following code:

```python
model_trainer = ModelTrainer(
training_image=image_uri, compute=compute, source_code=source_code,
hyperparameters=hyperparameters, environment=env,
base_job_name=job_prefix,
stopping_condition=StoppingCondition(max_runtime_in_seconds=90000),
checkpoint_config=CheckpointConfig(s3_uri=f"{checkpoint_s3_path}/{job_prefix}"),
)
model_trainer.fit(...)
```

**Expected behavior**
Training to start

**Screenshots or logs**

Image

**System information**
A description of your system. Please provide:
- **SageMaker Python SDK version**: 2.248.0
- **Framework name (eg. PyTorch) or algorithm (eg. KMeans)**: N/A
- **Framework version**: N/A
- **Python version**: 3.12
- **CPU or GPU**: GPU
- **Custom Docker image (Y/N)**: N

**Additional context**
Tried to downgrade to other SageMaker SDK versions, but couldn't get to a working one.

コントリビューションガイド

コントリビューションガイドを開く

調査の方向性

ModelTrainer API から始め、SageMaker Python SDK 2.248.0 と Python 3.12 を使用して、提供された fit(...) の例を再現します。SageMaker セッションの Pydantic 検証を追跡し、どの有効なセッション入力が受け入れられるかを特定します。完了条件は、ModelTrainer.fit(...) が検証エラーなしでトレーニングを開始することです。

索引モデルが issue の本文から書いたものです。

評価

技術スタック
aws, python
領域
api, cloud, machine-learning
issue の種類
バグ
難易度
3/5
見積もり時間
1〜2日
活発さ
停滞
明瞭さ
おおむね明確
初心者へのやさしさ
42/100

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。