bot commited on
Commit ·
0f1e257
1
Parent(s): d1dd088
Add repo to PYTHONPATH for so100_dataset import
Browse files- train_cloud.sh +1 -0
train_cloud.sh
CHANGED
|
@@ -29,6 +29,7 @@ if [ "$NUM_GPUS" -gt 1 ]; then
|
|
| 29 |
ACCEL_FLAGS="--multi_gpu --num_processes $NUM_GPUS"
|
| 30 |
fi
|
| 31 |
|
|
|
|
| 32 |
accelerate launch $ACCEL_FLAGS \
|
| 33 |
-m lerobot.scripts.lerobot_train \
|
| 34 |
--dataset.repo_id="so100:$DATASET_DIR:/workspace/pi05-so100-diverse/filtered_index.json:/workspace/pi05-so100-diverse/norm_stats.json" \
|
|
|
|
| 29 |
ACCEL_FLAGS="--multi_gpu --num_processes $NUM_GPUS"
|
| 30 |
fi
|
| 31 |
|
| 32 |
+
PYTHONPATH=/workspace/pi05-so100-diverse:$PYTHONPATH \
|
| 33 |
accelerate launch $ACCEL_FLAGS \
|
| 34 |
-m lerobot.scripts.lerobot_train \
|
| 35 |
--dataset.repo_id="so100:$DATASET_DIR:/workspace/pi05-so100-diverse/filtered_index.json:/workspace/pi05-so100-diverse/norm_stats.json" \
|