Upload policy weights, train config and readme

README.md

@@ -1,13 +1,13 @@
 ---
-datasets: ozgraslan/pusht_224
+datasets: ozgraslan/pusht_224_new
 library_name: lerobot
 license: apache-2.0
 model_name: diffusion_dino
 pipeline_tag: robotics
 tags:
+- robotics
 - lerobot
 - diffusion_dino
-- robotics
 ---
 
 # Model Card for diffusion_dino

config.json

@@ -13,7 +13,7 @@
         "observation.state": {
             "type": "STATE",
             "shape": [
-                7
+                4
             ]
         }
     },

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:68b4ac7fc7252251020443eb2dfcc1646ad84ec316beb2e277bf670dead82135
-size 1095528640
+oid sha256:331886cdf257c30a036e1eef3667e54a6fa54645dfde29f8a979ef089c5843fb
+size 1094840512

train_config.json

@@ -1,6 +1,6 @@
 {
     "dataset": {
-        "repo_id": "ozgraslan/pusht_224",
+        "repo_id": "ozgraslan/pusht_224_new",
         "root": null,
         "episodes": null,
         "image_transforms": {
@@ -90,10 +90,10 @@
                     2
                 ]
             },
-            "state": {
+            "proprio": {
                 "type": "STATE",
                 "shape": [
-                    7
+                    4
                 ]
             },
             "pixels": {
@@ -107,16 +107,16 @@
         },
         "features_map": {
             "action": "action",
-            "state_full": "observation.state",
+            "proprio": "observation.state",
             "pixels": "observation.image"
         },
         "max_parallel_tasks": 1,
         "disable_env_checker": true,
         "episode_length": 300,
-        "obs_type": "pixels_state",
+        "obs_type": "pixels_agent_pos_vel",
         "render_mode": "rgb_array",
-        "visualization_width": 680,
-        "visualization_height": 680,
+        "visualization_width": 768,
+        "visualization_height": 768,
         "observation_height": 224,
         "observation_width": 224
     },
@@ -135,7 +135,7 @@
             "observation.state": {
                 "type": "STATE",
                 "shape": [
-                    7
+                    4
                 ]
             }
         },
@@ -203,7 +203,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "/network/scratch/o/ozgur.aslan/outputs/train/diffusion_dino_pusht_1",
+    "output_dir": "/network/scratch/o/ozgur.aslan/outputs/train/diffusion_dino_pusht_4",
     "job_name": "pusht_224_diffusion_dino",
     "resume": false,
     "seed": 100000,
@@ -242,7 +242,7 @@
         "project": "lerobot-diffusion-pusht",
         "entity": "jakd9",
         "notes": null,
-        "run_id": "af1odhnx",
+        "run_id": "jb6kvdk3",
         "mode": null
     },
     "checkpoint_path": null,