File size: 4,454 Bytes
319eb16 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | # GVL
export GEMINI_API_KEY="your-api-key-here"
uv run python robometer/evals/run_baseline_eval.py \
reward_model=gvl \
model_config.model_name=gemini-2.5-flash-lite \
custom_eval.eval_types=[confusion_matrix] \
custom_eval.confusion_matrix=[[aliangdw_usc_franka_policy_ranking_usc_franka_policy_ranking,jesbu1_utd_so101_clean_policy_ranking_top_utd_so101_clean_policy_ranking_top,aliangdw_usc_xarm_policy_ranking_usc_xarm_policy_ranking]] \
max_frames=8
uv run python robometer/evals/run_baseline_eval.py \
reward_model=gvl \
model_config.provider=openai \
model_config.model_name=gpt-4o-mini \
custom_eval.eval_types=[confusion_matrix] \
custom_eval.confusion_matrix=[[aliangdw_usc_franka_policy_ranking_usc_franka_policy_ranking,jesbu1_utd_so101_clean_policy_ranking_top_utd_so101_clean_policy_ranking_top,aliangdw_usc_xarm_policy_ranking_usc_xarm_policy_ranking]] \
max_frames=8
# ReWIND
uv run python robometer/evals/run_baseline_eval.py \
reward_model=rewind \
model_path=rewardfm/rewind-scale-rfm1M-32layers-8frame-20260118-180522 \
custom_eval.eval_types=[confusion_matrix] \
custom_eval.confusion_matrix=[[aliangdw_usc_franka_policy_ranking_usc_franka_policy_ranking,jesbu1_utd_so101_clean_policy_ranking_top_utd_so101_clean_policy_ranking_top,aliangdw_usc_xarm_policy_ranking_usc_xarm_policy_ranking]] \
custom_eval.use_frame_steps=false \
max_frames=8 \
model_config.batch_size=64
# Robo-Dopamine (run with venv Python so vLLM is found; do not use uv run)
.venv-robodopamine/bin/python robometer/evals/run_baseline_eval.py \
reward_model=robodopamine \
model_path=tanhuajie2001/Robo-Dopamine-GRM-3B \
custom_eval.eval_types=[confusion_matrix] \
custom_eval.confusion_matrix=[[aliangdw_usc_franka_policy_ranking_usc_franka_policy_ranking,jesbu1_utd_so101_clean_policy_ranking_top_utd_so101_clean_policy_ranking_top,aliangdw_usc_xarm_policy_ranking_usc_xarm_policy_ranking]] \
max_frames=64 \
model_config.batch_size=1
# VLAC
uv run --extra vlac --python .venv-vlac/bin/python python robometer/evals/run_baseline_eval.py \
reward_model=vlac \
model_path=InternRobotics/VLAC \
custom_eval.eval_types=[confusion_matrix] \
custom_eval.confusion_matrix=[[aliangdw_usc_franka_policy_ranking_usc_franka_policy_ranking,jesbu1_utd_so101_clean_policy_ranking_top_utd_so101_clean_policy_ranking_top,aliangdw_usc_xarm_policy_ranking_usc_xarm_policy_ranking]] \
custom_eval.pad_frames=false \
max_frames=64
# RoboReward-8B
# without koch
uv run python robometer/evals/run_baseline_eval.py \
reward_model=roboreward \
model_path=teetone/RoboReward-8B \
custom_eval.eval_types=[confusion_matrix] \
custom_eval.confusion_matrix=[[aliangdw_usc_franka_policy_ranking_usc_franka_policy_ranking,jesbu1_utd_so101_clean_policy_ranking_top_utd_so101_clean_policy_ranking_top,aliangdw_usc_xarm_policy_ranking_usc_xarm_policy_ranking]] \
max_frames=64
# on all
uv run python robometer/evals/run_baseline_eval.py \
reward_model=roboreward \
model_path=teetone/RoboReward-8B \
custom_eval.eval_types=[confusion_matrix] \
custom_eval.confusion_matrix=[[aliangdw_usc_franka_policy_ranking_usc_franka_policy_ranking,jesbu1_utd_so101_clean_policy_ranking_top_utd_so101_clean_policy_ranking_top,aliangdw_usc_xarm_policy_ranking_usc_xarm_policy_ranking,jesbu1_usc_koch_p_ranking_rfm_usc_koch_p_ranking_all]] \
max_frames=64
# Robometer-4B
# without koch
uv run python robometer/evals/run_baseline_eval.py \
reward_model=rbm \
model_path=aliangdw/Robometer-4B \
custom_eval.eval_types=[confusion_matrix] \
custom_eval.confusion_matrix=[[aliangdw_usc_franka_policy_ranking_usc_franka_policy_ranking,jesbu1_utd_so101_clean_policy_ranking_top_utd_so101_clean_policy_ranking_top,aliangdw_usc_xarm_policy_ranking_usc_xarm_policy_ranking]] \
max_frames=8 \
model_config.batch_size=32
# on all
uv run python robometer/evals/run_baseline_eval.py \
reward_model=rbm \
model_path=aliangdw/Robometer-4B \
custom_eval.eval_types=[confusion_matrix] \
custom_eval.confusion_matrix=[[aliangdw_usc_franka_policy_ranking_usc_franka_policy_ranking,jesbu1_utd_so101_clean_policy_ranking_top_utd_so101_clean_policy_ranking_top,aliangdw_usc_xarm_policy_ranking_usc_xarm_policy_ranking,jesbu1_usc_koch_p_ranking_rfm_usc_koch_p_ranking_all]] \
max_frames=8 \
model_config.batch_size=32 |