AWS Machine Learning Blog· Nicolas Jourdan·· 3 天前AI 评分58
Amazon SageMaker AI 推出训练会话 API,可自写训练循环而无需管理 GPU
Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
AI 导读
Amazon SageMaker AI 推出训练会话 API,把抽象层下移到训练操作,开发者用 forward_backward()、optim_step()、sample() 等调用在本地 Python 进程里驱动托管 GPU 上的有状态会话,服务负责 GPU 分配、模型与 LoRA 加载、分布式执行、故障恢复和检查点存储。
来源:AWS Machine Learning Blog · aws.amazon.com