[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
2
.gitignore
vendored
2
.gitignore
vendored
@@ -173,4 +173,4 @@ dmypy.json
|
||||
cython_debug/
|
||||
|
||||
# for mac
|
||||
._*
|
||||
._*
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user