Update README.md
Browse files
README.md
CHANGED
|
@@ -12,6 +12,7 @@ tags:
|
|
| 12 |
- aloha
|
| 13 |
- imitation-learning
|
| 14 |
- benchmark
|
|
|
|
| 15 |
---
|
| 16 |
|
| 17 |
# 🦾 Diffusion Policy for Aloha Insertion (200k Steps)
|
|
@@ -70,6 +71,7 @@ For reproducibility, here are the key parameters used during the training sessio
|
|
| 70 |
|
| 71 |
<details>
|
| 72 |
<summary>📄 <strong>Click to view full <code>diffusion_aloha.yaml</code> used for training</strong></summary>
|
|
|
|
| 73 |
|
| 74 |
```yaml
|
| 75 |
# @package _global_
|
|
@@ -128,7 +130,7 @@ policy:
|
|
| 128 |
# --- 噪声调度器 (DDPM) ---
|
| 129 |
noise_scheduler_type: DDPM
|
| 130 |
num_train_timesteps: 100
|
| 131 |
-
|
| 132 |
beta_schedule: squaredcos_cap_v2
|
| 133 |
beta_start: 0.0001
|
| 134 |
beta_end: 0.02
|
|
@@ -145,9 +147,7 @@ policy:
|
|
| 145 |
scheduler_warmup_steps: 500
|
| 146 |
|
| 147 |
use_amp: true
|
| 148 |
-
|
| 149 |
-
|
| 150 |
-
\</details\>
|
| 151 |
|
| 152 |
-----
|
| 153 |
|
|
@@ -175,7 +175,4 @@ python -m lerobot.scripts.lerobot_eval \
|
|
| 175 |
--eval.batch_size 8 \
|
| 176 |
--env.type aloha \
|
| 177 |
--env.task AlohaInsertion-v0
|
| 178 |
-
```
|
| 179 |
-
|
| 180 |
-
```
|
| 181 |
```
|
|
|
|
| 12 |
- aloha
|
| 13 |
- imitation-learning
|
| 14 |
- benchmark
|
| 15 |
+
|
| 16 |
---
|
| 17 |
|
| 18 |
# 🦾 Diffusion Policy for Aloha Insertion (200k Steps)
|
|
|
|
| 71 |
|
| 72 |
<details>
|
| 73 |
<summary>📄 <strong>Click to view full <code>diffusion_aloha.yaml</code> used for training</strong></summary>
|
| 74 |
+
</details>
|
| 75 |
|
| 76 |
```yaml
|
| 77 |
# @package _global_
|
|
|
|
| 130 |
# --- 噪声调度器 (DDPM) ---
|
| 131 |
noise_scheduler_type: DDPM
|
| 132 |
num_train_timesteps: 100
|
| 133 |
+
num_inference_timesteps: 100
|
| 134 |
beta_schedule: squaredcos_cap_v2
|
| 135 |
beta_start: 0.0001
|
| 136 |
beta_end: 0.02
|
|
|
|
| 147 |
scheduler_warmup_steps: 500
|
| 148 |
|
| 149 |
use_amp: true
|
| 150 |
+
```
|
|
|
|
|
|
|
| 151 |
|
| 152 |
-----
|
| 153 |
|
|
|
|
| 175 |
--eval.batch_size 8 \
|
| 176 |
--env.type aloha \
|
| 177 |
--env.task AlohaInsertion-v0
|
|
|
|
|
|
|
|
|
|
| 178 |
```
|