獲得模型權重文件
Shell
cd ~/output_lerobot_train/a/checkpoints/last
zip -r ~/ckpt.zip pretrained_model每20K個step,保存一次模型權重文件
右鍵下載ckpt.zip壓縮包,解壓到本地電腦
握手
Shell
# 指定訓練步數保存的模型
# cd ~/output_lerobot_train/shake/diffusion/checkpoints/005000
# 最新的模型
cd ~/output_lerobot_train/shake/wallx/checkpoints/last
zip -r ~/ckpt.zip pretrained_model
