mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-23 09:46:00 +00:00
remove sh files
This commit is contained in:
@@ -1,53 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
# config
|
|
||||||
REPO_ID=jadechoghari/smol-libero3
|
|
||||||
TASK=libero_10,libero_spatial
|
|
||||||
OUTPUT_DIR=./outputs/
|
|
||||||
|
|
||||||
# clean previous run
|
|
||||||
rm -rf $OUTPUT_DIR
|
|
||||||
|
|
||||||
# training params
|
|
||||||
STEPS=100000
|
|
||||||
BATCH_SIZE=4
|
|
||||||
EVAL_FREQ=1
|
|
||||||
SAVE_FREQ=10000
|
|
||||||
NUM_WORKERS=4
|
|
||||||
|
|
||||||
# model params
|
|
||||||
POLICY=smolvla
|
|
||||||
USE_AMP=false
|
|
||||||
OPTIMIZER_LR=1e-4
|
|
||||||
PEFT_METHOD=lora
|
|
||||||
LOAD_VLM_WEIGHTS=true
|
|
||||||
VLM_REPO_ID=None
|
|
||||||
MAX_ACTION_DIM=32
|
|
||||||
MAX_STATE_DIM=32
|
|
||||||
|
|
||||||
# dataset/image params
|
|
||||||
USE_IMAGENET_STATS=false
|
|
||||||
ENABLE_IMG_TRANSFORM=true
|
|
||||||
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
|
|
||||||
python src/lerobot/scripts/train.py \
|
|
||||||
--policy.type=$POLICY \
|
|
||||||
--dataset.repo_id=$REPO_ID \
|
|
||||||
--env.type=libero \
|
|
||||||
--env.task=$TASK \
|
|
||||||
--output_dir=$OUTPUT_DIR \
|
|
||||||
--steps=$STEPS \
|
|
||||||
--batch_size=$BATCH_SIZE \
|
|
||||||
--eval_freq=$EVAL_FREQ \
|
|
||||||
--save_freq=$SAVE_FREQ \
|
|
||||||
--num_workers=$NUM_WORKERS \
|
|
||||||
--policy.repo_id=$VLM_REPO_ID \
|
|
||||||
--env.multitask_eval=True \
|
|
||||||
--eval.batch_size=1 \
|
|
||||||
--eval.n_episodes=1 \
|
|
||||||
@@ -1,19 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
unset LEROBOT_HOME
|
|
||||||
unset HF_LEROBOT_HOME
|
|
||||||
# CONFIGURATION
|
|
||||||
POLICY_PATH="bicmol/smolvla-libero"
|
|
||||||
TASK=libero_spatial
|
|
||||||
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=False \
|
|
||||||
--env.task=$TASK \
|
|
||||||
Reference in New Issue
Block a user