跳到正文
AWS Machine Learning Blog· Nicolas Jourdan·· 18 天前精选AI 评分82

Amazon SageMaker AI 发布训练会话 API,支持无 GPU 管理自定义训练循环

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

AI 导读

Amazon SageMaker AI 发布训练会话 API,允许用户在本地 Python 进程中编写训练循环,由服务托管 GPU 执行采样、梯度更新和检查点操作。该功能支持 SFT、RL 等自定义算法,按 token 计费,并配套发布包含 SFT 和 RL 配方的训练会话 cookbook。

推荐理由

该功能将抽象边界下沉至训练操作层,允许用户在本地编写训练循环,由服务托管 GPU 基础设施,从而在算法自由度与运维成本间取得平衡。

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