final refactor/fix

2026-05-16 17:20:05 +00:00 · 2025-08-25 06:25:02 -04:00
parent afad90ffaa
commit 8d2c66abd2
7 changed files with 47 additions and 75 deletions
@@ -1,7 +1,7 @@
 #!/bin/bash

 # config
-REPO_ID=yzembodied/libero_10_image_task_1
+REPO_ID=jadechoghari/smol-libero
 TASK=libero_10
 OUTPUT_DIR=./outputs/

@@ -2,14 +2,12 @@

 unset LEROBOT_HOME
 unset HF_LEROBOT_HOME
-# === CONFIGURATION ===
-POLICY_PATH="ganatrask/lerobot-pi0-libero-object"  # or outputs/train/.../pretrained_model
+# CONFIGURATION
+POLICY_PATH="ganatrask/lerobot-pi0-libero-object"
 TASK=libero_object
 ENV_TYPE="libero"
 BATCH_SIZE=1
 N_EPISODES=1
-USE_AMP=false
-DEVICE=cuda

 # RUN EVALUATION
 python src/lerobot/scripts/eval.py \