第七步:取得模型權重檔案
在雲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
