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 等调用自行编写训练循环,GPU 分配与分布式执行由服务托管。
来源:AWS Machine Learning Blog · aws.amazon.com