跳到正文
原文
AWS Machine Learning Blog· Nicolas Jourdan·· 3 小时前AI 评估 · 68/100分数表示内容值得关注的程度,依据实质份量、信息增量、证据强度、读者相关性和可用性,按内容类型加权。显示分数为两次独立评分的平均值。AI 判断可能有误,评分不代表事实已被核实,也不同于热点榜的讨论热度。

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

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

AI 导读

Amazon SageMaker AI 的 Serverless 模型定制新增训练会话 API,把 forward_backward()、optim_step()、sample() 等操作变成 SageMaker API 调用,模型在 GPU 上保持加载,用户只需在本地 Python 进程里编写训练循环。

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