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 推出训练会话 API，把 forward_backward、optim_step、sample 等操作变成 API 调用，模型在托管 GPU 上保持加载，用户只需在本地 Python 中编写训练循环。
来源：AWS Machine Learning Blog · aws.amazon.com