diff --git a/.gitignore b/.gitignore index 7cff9f2e..7f3377e1 100644 --- a/.gitignore +++ b/.gitignore @@ -173,4 +173,4 @@ dmypy.json cython_debug/ # for mac -._* \ No newline at end of file +._* diff --git a/examples/train_smolvla.sh b/examples/train_smolvla.sh index 8c06aa45..6c4cf956 100644 --- a/examples/train_smolvla.sh +++ b/examples/train_smolvla.sh @@ -22,14 +22,14 @@ ##SBATCH -C a100 ###SBATCH -A dyf@a100 ##SBATCH -A lqm@a100 -##SBATCH --qos=qos_gpu_a100-dev -##SBATCH --qos=qos_gpu_a100-t3 +##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 +##SBATCH --qos=qos_gpu_h100-dev +#SBATCH --qos=qos_gpu_h100-t4 cd ~/lerobot_pi @@ -45,7 +45,7 @@ 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_DATASETS_OFFLINE=1 export HF_HUB_OFFLINE=1 export WANDB_CACHE_DIR=/lustre/fsn1/projects/rech/dyf/ugz83ue/wandb @@ -103,4 +103,3 @@ python lerobot/scripts/train.py \ --output_dir=$TRAIN_DIR \ --batch_size=$BATCH_SIZE \ --steps=$OFFLINE_STEPS - diff --git a/lerobot/common/robot_devices/control_utils.py b/lerobot/common/robot_devices/control_utils.py index 103c4fdb..60656196 100644 --- a/lerobot/common/robot_devices/control_utils.py +++ b/lerobot/common/robot_devices/control_utils.py @@ -256,7 +256,7 @@ def control_loop( else: observation = robot.capture_observation() action = None - observation['task'] = [single_task] + observation["task"] = [single_task] observation["robot_type"] = [policy.robot_type] if hasattr(policy, "robot_type") else [""] if policy is not None: pred_action = predict_action(