AWS Machine Learning Blog· Nicolas Jourdan·2026-10-23 02:00· 4 小时前精选AI 评分60

Amazon SageMaker AI 推出训练会话 API，自写训练循环无需管理 GPU

Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs

AI 导读

Amazon SageMaker AI 推出训练会话 API，把 forward_backward()、optim_step()、sample() 等操作变成 API 调用，模型在托管 GPU 上保持加载，用户只需在本地 Python 进程里编写训练循环。

推荐理由

原文给出训练会话 API 的编程模型与两个可运行示例，读者可据此判断自定义训练循环与托管 GPU 的边界如何划分。

来源：AWS Machine Learning Blog · aws.amazon.com