Libero-已有模型推理#
本章节介绍如何使用已训练好的模型在 Libero 任务上进行推理评估。
准备工作#
完成环境安装(参考 :doc:
installation/index)。
快速开始#
首先,准备好需要评测的模型权重,或从 HuggingFace 下载预训练权重。 然后设置环境变量,需要根据实际GPU数量修改MLP_WORKER_GPU:
export MLP_WORKER_GPU=8
export MLP_WORKER_NUM=1
export MLP_ROLE_INDEX=0
export MLP_WORKER_0_HOST=localhost
export MLP_WORKER_0_PORT=29500
运行 scripts/eval.sh 进行LIBERO-10评估:
bash scripts/eval.sh \
configs/pi05/pi05_paligemma_libero10_full_finetune.py \
work_dirs/pi05_paligemma_libero10_full_finetune/checkpoint_step_10000.pt # 替换为您下载的checkpoint路径
结果查看#
评估完成后可在checkpoint目录中查看:
评估日志(成功率、完成率等指标)
rollouts/视频
更多参数细节请参考 :doc:vla-eval。