Upload policy weights, train config and readme

README.md

@@ -5,9 +5,9 @@ license: apache-2.0
 model_name: diffusion
 pipeline_tag: robotics
 tags:
+- diffusion
 - lerobot
 - robotics
-- diffusion
 ---
 
 # Model Card for diffusion

config.json

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

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3f46245c2c472d2830a5ef625441362baa2533dc5856573892905542680652ad
-size 356574368
+oid sha256:b0f2859dc9f6329a3d35a473b4e86ea15c3979a326fc2828996ff4924e0281ff
+size 1110597968

train_config.json

@@ -126,9 +126,9 @@
         "spatial_softmax_num_keypoints": 32,
         "use_separate_rgb_encoder_per_camera": true,
         "down_dims": [
-            256,
             512,
-            1024
+            1024,
+            2048
         ],
         "kernel_size": 5,
         "n_groups": 8,
@@ -154,7 +154,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "/scratch/ameypore/runs/lerobot/1298743/checkpoints",
+    "output_dir": "/scratch/ameypore/runs/lerobot/1298744/checkpoints",
     "job_name": "diffusion",
     "resume": false,
     "seed": 1000,