AWS Machine Learning Blog· Nicolas Jourdan·· 2 小时前同新闻AI 评分68
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 上执行模型采样、梯度计算和检查点操作。用户可组合这些 API 实现 SFT、偏好优化或 RL，服务负责 GPU 分配、分布式执行和故障恢复。文章展示了 GSM8K 和竞赛编程任务的 RL 示例，并报告了各自训练后的准确率变化。
同一新闻，精选展示《AWS SageMaker AI 推出无服务器模型定制训练会话》
来源：AWS Machine Learning Blog · aws.amazon.com