AWS Machine Learning Blog· Nicolas Jourdan·· 4 小时前AI 评分55
Amazon SageMaker AI 推出 training session API，可自写训练循环而无需自管 GPU
Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
AI 导读
AWS 在 Amazon SageMaker AI 的 serverless 模型定制中推出 training session API，把 forward_backward()、optim_step()、sample() 等操作变成面向有状态 GPU 会话的 API 调用，用户在本地写 Python 训练循环，服务侧负责分布式执行、故障恢复和检查点管理。
来源：AWS Machine Learning Blog · aws.amazon.com