feat: add pusht dit no-causal config

2026-03-27 17:06:16 +08:00
parent 36fbf2a6b7
commit 31925bbf39
3 changed files with 96 additions and 0 deletions
--- a/image_pusht_diffusion_policy_dit_nocausal.yaml
+++ b/image_pusht_diffusion_policy_dit_nocausal.yaml
@@ -0,0 +1,31 @@
+defaults:
+  - diffusion_policy/config/train_diffusion_transformer_hybrid_workspace@_here_
+  - override /diffusion_policy/config/task@task: pusht_image
+  - _self_
+
+exp_name: pusht_image_dit_nocausal
+
+policy:
+  _target_: diffusion_policy.policy.diffusion_transformer_hybrid_image_policy.DiffusionTransformerHybridImagePolicy
+  causal_attn: false
+
+logging:
+  backend: swanlab
+  mode: online
+  name: ${exp_name}
+  resume: false
+  tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
+  id: null
+  group: ${exp_name}
+
+dataloader:
+  num_workers: 0
+
+val_dataloader:
+  num_workers: 0
+
+task:
+  env_runner:
+    n_envs: 1
+    n_test_vis: 0
+    n_train_vis: 0