跳至主要内容

模型訓練

訓練頁為整體流程中欄位數量最多的頁面。實際顯示的欄位數取決於所選的策略:介面依策略調整,僅保留對該策略確實生效的參數。

訓練頁面

啟動訓練前的四項設定​

一、資料集選擇。 選定本次訓練使用的資料集。實體錄製與模擬生成的資料在此處無區別,兩者均為 LeRobot 標準格式並列於同一清單。若曾執行合併作業,應選取合併後的資料集。

二、策略選擇(ACT / GR00T N1.5)。 此為策略選擇所述的分岔點,同時決定了後續可用的部署硬體。

三、選擇裝置。 訓練頁的裝置清單依主機配置而異:具備 NVIDIA 顯示卡時提供 GPU (CUDA) 與 CPU,否則提供 GPU (OpenVINO) 與 CPU。介面提示已載明 GPU 的運算速度顯著高於 CPU,故在具備 GPU 的環境下應選用 GPU。

備註

GPU (TensorRT) 僅在推論頁出現,因其屬於部署階段的設定。於訓練頁未見該選項屬正常情況。

四、輸出資料夾名稱。 指定訓練產物的儲存位置。停止輸入約一秒後,平台將自動檢查該名稱是否已被使用,並於欄位下方顯示檢查結果(可用或已存在)。

由於訓練作業的執行時間以小時計,使用既有名稱可能導致既有訓練成果被覆寫,且該情況通常須至訓練完成後方能察覺。建議於命名時納入可識別的資訊,例如任務名稱、策略與版本。


下一步: 訓練參數.