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

Amazon SageMaker AI 推出训练会话 API,无需管理 GPU 即可编写自定义训练循环

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

AI 导读

Amazon SageMaker AI 推出训练会话(Training Session)API,用户可在本地 Python 环境中直接驱动远程 GPU 会话,通过 sample、forward_backward、optim_step 等操作自主编排训练循环,服务负责模型加载、分布式执行、故障恢复与 checkpoint 存储。

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