add safethread support

This commit is contained in:
Jade Choghari (jchoghar)
2025-08-25 14:52:35 -04:00
parent 4be3942cbc
commit ff861ba869
3 changed files with 43 additions and 25 deletions
+2 -2
View File
@@ -32,7 +32,7 @@ MAX_NUM_IMAGES=2
MAX_IMAGE_DIM=1024
unset LEROBOT_HOME
unset HF_LEROBOT_HOME
export MUJOCO_GL=egl
echo -e "\033[1;33m[WARNING]\033[0m LIBERO is not yet fully supported in this PR!"
# launch
@@ -48,6 +48,6 @@ python src/lerobot/scripts/train.py \
--save_freq=$SAVE_FREQ \
--num_workers=$NUM_WORKERS \
--policy.repo_id=$VLM_REPO_ID \
--env.multitask_eval=True \
--env.multitask_eval=False \
--eval.batch_size=1 \
--eval.n_episodes=1 \
+2 -2
View File
@@ -8,12 +8,12 @@ TASK=libero_object
ENV_TYPE="libero"
BATCH_SIZE=1
N_EPISODES=1
export MUJOCO_GL=egl
# RUN EVALUATION
python src/lerobot/scripts/eval.py \
--policy.path="$POLICY_PATH" \
--env.type="$ENV_TYPE" \
--eval.batch_size="$BATCH_SIZE" \
--eval.n_episodes="$N_EPISODES" \
--env.multitask_eval=True \
--env.multitask_eval=False \
--env.task=$TASK \