hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

clean train

Browse Source
This commit is contained in:
mshukor
2025-05-30 13:30:44 +02:00
parent ac0f2e9dd9
commit 8097f66248
1 changed files with 1 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
examples/train_smolvla.sh
View File

@@ -1,71 +1,12 @@
#!/bin/bash
#SBATCH --job-name=lerobot_smolvla_test_task1_main_pretrained
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --gpus-per-node=1
#SBATCH --mail-type=END,FAIL
#SBATCH --output=/lustre/fswork/projects/rech/dyf/ugz83ue/logs/slurm/lerobot_smolvla_test_task1_main_pretrained.out
###SBATCH --nodelist=jean-zay-a101
#SBATCH --cpus-per-task=20
###SBATCH --exclusive
#SBATCH --time=40:00:00
#SBATCH --mail-user=mustafa.shukor@isir.upmc.fr
##SBATCH --partition=gpu_p2
##SBATCH --qos=qos_gpu-t3
###SBATCH -C v100-32g
##SBATCH -A dyf@v100
##SBATCH --partition=gpu_p5
##SBATCH -C a100
###SBATCH -A dyf@a100
##SBATCH -A lqm@a100
##SBATCH --qos=qos_gpu_a100-dev
##SBATCH --qos=qos_gpu_a100-t3
#SBATCH --partition=gpu_p6
#SBATCH -C h100
#SBATCH -A lqm@h100
##SBATCH --qos=qos_gpu_h100-dev
#SBATCH --qos=qos_gpu_h100-t4
cd ~/lerobot_pi
cd ~/lerobot
source ~/.bashrc
source activate lerobot_main
export LC_ALL=C
rm core-*
export TRANSFORMERS_CACHE=$WORK/.cache/huggingface/transformers
export HF_HOME=$WORK/.cache/huggingface
export DATA_DIR=$WORK/.cache/huggingface/datasets
export HF_LEROBOT_HOME=$WORK/.cache/huggingface/lerobot
# export LEROBOT_HOME=
export HF_DATASETS_OFFLINE=1
export HF_HUB_OFFLINE=1
export WANDB_CACHE_DIR=/lustre/fsn1/projects/rech/dyf/ugz83ue/wandb
export WANDB_MODE=offline
export TOKENIZERS_PARALLELISM=false
cd ~/lerobot_pi
# ###### dgx
# source ~/miniconda3/bin/activate
# conda activate lerobot
# export WORK=/home/mustafa_shukor
# V3 So100
REPO_ID=danaaubakirova/svla_so100_task1_v3
DATASET_NAME=so100_v3_task_1
@@ -79,19 +20,6 @@ OFFLINE_STEPS=200000
BATCH_SIZE=64
# TASK_NAME=lerobot_${DATASET_NAME}_${POLICY_NAME}
# TRAIN_DIR=$WORK/logs/lerobot/$TASK_NAME
# echo $TRAIN_DIR
# rm -r $TRAIN_DIR
# python lerobot/scripts/train.py \
# --policy.type=$POLICY \
# --dataset.repo_id=$REPO_ID \
# --output_dir=$TRAIN_DIR \
# --batch_size=$BATCH_SIZE \
# --steps=$OFFLINE_STEPS
TASK_NAME=lerobot_${DATASET_NAME}_${POLICY_NAME}_pretrained
TRAIN_DIR=$WORK/logs/lerobot/$TASK_NAME
echo $TRAIN_DIR