Upload policy weights, train config and readme

config.json

@@ -52,9 +52,9 @@
     "spatial_softmax_num_keypoints": 32,
     "use_separate_rgb_encoder_per_camera": false,
     "down_dims": [
+        256,
         512,
-        1024,
-        2048
+        1024
     ],
     "kernel_size": 5,
     "n_groups": 8,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d2a801bab63466a02aa62c8738a028b2d8f2d93370170626ecbab10b1ff9fae4
-size 1050861448
+oid sha256:12e94a1571b2dc21467cbff846ed196353ca719a0b96310ee1234fe6eb29c61b
+size 304306928

train_config.json

@@ -175,9 +175,9 @@
         "spatial_softmax_num_keypoints": 32,
         "use_separate_rgb_encoder_per_camera": false,
         "down_dims": [
+            256,
             512,
-            1024,
-            2048
+            1024
         ],
         "kernel_size": 5,
         "n_groups": 8,
@@ -203,7 +203,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/train/2026-01-17/02-02-18_pusht_diffusion",
+    "output_dir": "outputs/train/2026-01-17/09-12-16_pusht_diffusion",
     "job_name": "pusht_diffusion",
     "resume": false,
     "seed": 100000,
@@ -243,7 +243,7 @@
         "project": "u-net",
         "entity": null,
         "notes": null,
-        "run_id": "y9zx3s6p",
+        "run_id": "r6z607tw",
         "mode": null
     },
     "use_rabc": false,