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