Update README.md
Browse files
README.md
CHANGED
|
@@ -69,6 +69,20 @@ For reproducibility, here are the key parameters used during the training sessio
|
|
| 69 |
- **Scheduler**: Cosine with warmup
|
| 70 |
- **Vision**: ResNet18 with GroupNorm (Cropped to 420x560)
|
| 71 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 72 |
<details>
|
| 73 |
<summary>📄 <strong>Click to view full <code>diffusion_aloha.yaml</code> used for training</strong></summary>
|
| 74 |
|
|
|
|
| 69 |
- **Scheduler**: Cosine with warmup
|
| 70 |
- **Vision**: ResNet18 with GroupNorm (Cropped to 420x560)
|
| 71 |
|
| 72 |
+
### Original Training Command (My Resume Mode)
|
| 73 |
+
|
| 74 |
+
```bash
|
| 75 |
+
python -m lerobot.scripts.lerobot_train \
|
| 76 |
+
--config_path diffusion_aloha.yaml \
|
| 77 |
+
--env.type aloha \
|
| 78 |
+
--env.task AlohaInsertion-v0 \
|
| 79 |
+
--dataset.repo_id lerobot/aloha_sim_insertion_human \
|
| 80 |
+
--wandb.enable true \
|
| 81 |
+
--job_name DP_Aloha_Insertion \
|
| 82 |
+
--policy.repo_id Lemon-03/DP_Aloha_Insertion_test \
|
| 83 |
+
```
|
| 84 |
+
|
| 85 |
+
### diffusion_aloha.yaml
|
| 86 |
<details>
|
| 87 |
<summary>📄 <strong>Click to view full <code>diffusion_aloha.yaml</code> used for training</strong></summary>
|
| 88 |
|