SageMaker AI 推出可自訂的無伺服器訓練工作階段

快訊 大型語言模型訓練基礎設施AI 代理基礎設施

  • SageMaker AI 的無伺服器模型客製化訓練工作階段已正式推出;使用者可自行撰寫訓練迴圈,AWS 則負責 GPU 配置、分散式執行、故障復原與 checkpoint 儲存。
  • 功能支援 SFT、DPO、RLVR、RLAIF 與多輪 RL,並可選擇 LoRA 或完整微調,也能串接不同技術。
  • 訓練迴圈與 reward function 在使用者環境執行;工作階段中的 sampler 產生回應,trainer 計算梯度並更新模型權重。
  • 費用按 SageMaker AI 處理的 token 計算;reward function 在使用者環境評分期間會暫停計費,且不需預留容量。

本站見解

相較於 SageMaker AI 既有的受管理多輪 RL 技術,training sessions 讓團隊自行定義 loss 與 reward 邏輯,同時免去 GPU 叢集維運。文章未提供 token 價格,也沒有與自管 GPU 成本的比較,因此目前無法估算實際預算差異。

來源

  1. Announcing new ways to customize models with SageMaker AI, without managing GPUs · AWS Machine Learning

本文摘要與見解由 AI 自動生成,事實以原文為準。 AI 內容政策