Lemon-03 commited on
Commit
a54ea26
·
verified ·
1 Parent(s): 53c0ec7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -7
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
- num_inference_steps: 100
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
  ```