AWS Machine Learning Blog· Nicolas Jourdan·· 3 小时前AI 评分58
Amazon SageMaker AI 推出训练会话 API，自写训练循环无需管理 GPU
Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
AI 导读
Amazon SageMaker AI 的 Serverless 模型定制新增训练会话 API，把 forward_backward()、optim_step()、sample() 等操作变成 SageMaker API 调用，模型在 GPU 上保持加载，用户只需在本地 Python 进程中编写训练循环。
来源：AWS Machine Learning Blog · aws.amazon.com