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