🧪 新教程持續更新中 · 由機械臂到傳感器,跟隨教程從零搭建
跳到主要內容

第七步:取得模型權重檔案 ​

在雲GPU上訓練完後,可以把模型打包下載到本地:

Shell
cd ~/output_lerobot_train/a/checkpoints/last
zip -r ~/ckpt.zip pretrained_model

每20K個step,儲存一次模型權重檔案

右鍵下載ckpt.zip壓縮包,解壓到本地電腦

如果模型還要傳給其他人、或者你打算換一台電腦推理,把它直接傳到 Hugging Face 會比打包下載更省事,見上傳模型到HuggingFace(可選)。

握手 ​

Shell
# 指定訓練步數儲存的模型
# cd ~/output_lerobot_train/shake/diffusion/checkpoints/005000

# 最新的模型
cd ~/output_lerobot_train/shake/wallx/checkpoints/last
zip -r ~/ckpt.zip pretrained_model