跳到正文
AWS Machine Learning Blog· Nicolas Jourdan·· 3 小时前AI 评分59

Amazon SageMaker 推出 training session API,可自写训练循环而无需管理 GPU

Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs

阅读原文

本站未展示全文,请前往来源网站阅读。

AI 导读

AWS 为 Amazon SageMaker AI 推出 training session API,把 forward_backward()、optim_step()、sample() 等操作变为 API 调用,用户在本地写 Python 训练循环,服务侧托管 GPU、分布式执行、故障恢复和 checkpoint。

来源:AWS Machine Learning Blog · aws.amazon.com