feat(pusht): add dual-head uv transformer

2026-03-17 17:05:02 +08:00
parent 42dc29a2cb
commit 9169e4d7e0
3 changed files with 46 additions and 6 deletions
--- a/image_pusht_diffusion_policy_dit_pmf.yaml
+++ b/image_pusht_diffusion_policy_dit_pmf.yaml
@@ -63,6 +63,7 @@ policy:
  n_emb: 256
  n_head: 4
  n_layer: 12
+  n_head_layers: 4
  n_obs_steps: 2
  n_time_tokens: 4
  noise_scale: 1.0