AWS 机器学习· Nicolas Jourdan·· 18 小时前精选AI 评分70
Amazon SageMaker AI 推出训练会话 API,支持无 GPU 管理自定义训练循环
Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
AI 导读
Amazon SageMaker AI 推出训练会话 API(training session APIs)及配套 cookbook,允许开发者在本地 Python 代码中完全掌控训练循环逻辑,底层由托管 GPU 执行模型加载、采样与梯度更新。
推荐理由
该功能将训练循环逻辑与底层算力解耦,开发者在本地编写自定义奖励或损失函数的同时,无需运维 GPU 集群且按处理 token 计费。
来源:AWS 机器学习 · aws.amazon.com