bot commited on
Commit ·
66cf9d1
1
Parent(s): 3e13241
Add config_path for resume
Browse files- train_cloud.sh +1 -0
train_cloud.sh
CHANGED
|
@@ -32,6 +32,7 @@ fi
|
|
| 32 |
python3.12 -m accelerate.commands.launch $ACCEL_FLAGS \
|
| 33 |
-m lerobot.scripts.lerobot_train \
|
| 34 |
--resume=true \
|
|
|
|
| 35 |
--dataset.repo_id="so100:$DATASET_DIR:/workspace/pi05-so100-diverse/filtered_index.json:/workspace/pi05-so100-diverse/norm_stats.json" \
|
| 36 |
--policy.path=lerobot/pi05_base \
|
| 37 |
--policy.train_expert_only=true \
|
|
|
|
| 32 |
python3.12 -m accelerate.commands.launch $ACCEL_FLAGS \
|
| 33 |
-m lerobot.scripts.lerobot_train \
|
| 34 |
--resume=true \
|
| 35 |
+
--config_path=/ephemeral/production_run/checkpoints/last/pretrained_model/train_config.json \
|
| 36 |
--dataset.repo_id="so100:$DATASET_DIR:/workspace/pi05-so100-diverse/filtered_index.json:/workspace/pi05-so100-diverse/norm_stats.json" \
|
| 37 |
--policy.path=lerobot/pi05_base \
|
| 38 |
--policy.train_expert_only=true \
|