mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-16 00:59:46 +00:00
add safethread support
This commit is contained in:
@@ -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 \
|
||||
|
||||
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user