mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 20:19:43 +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 \
|
||||
|
||||
Reference in New Issue
Block a user