Upload policy weights, train config and readme

README.md

@@ -1,14 +1,14 @@
 ---
 base_model: lerobot/smolvla_base
-datasets: aaron-ser/SO101-Dataset
+datasets: astro189/record_scene_1
 library_name: lerobot
 license: apache-2.0
 model_name: smolvla
 pipeline_tag: robotics
 tags:
 - robotics
-- lerobot
 - smolvla
+- lerobot
 ---
 
 # Model Card for smolvla

config.json

@@ -13,12 +13,20 @@
                 6
             ]
         },
-        "observation.images.front": {
+        "observation.images.gripper": {
             "type": "VISUAL",
             "shape": [
                 3,
-                720,
-                1280
+                480,
+                640
+            ]
+        },
+        "observation.images.top": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
             ]
         }
     },
@@ -30,7 +38,7 @@
             ]
         }
     },
-    "device": "mps",
+    "device": "cuda",
     "use_amp": false,
     "push_to_hub": true,
     "repo_id": "aaron-ser/smolvla-model",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:48452c589475a9ed880a83b815275da6996a29f627bac7ee34b51fb8daa1d7db
+oid sha256:935a1f43b1d5e0296707d385a99f439fc417008c9f1b410851949291c3b9d5b6
 size 1197790016

train_config.json

@@ -1,8 +1,60 @@
 {
     "dataset": {
-        "repo_id": "aaron-ser/SO101-Dataset",
+        "repo_id": "astro189/record_scene_1",
         "root": null,
-        "episodes": null,
+        "episodes": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15,
+            16,
+            17,
+            18,
+            19,
+            20,
+            21,
+            22,
+            23,
+            24,
+            25,
+            26,
+            27,
+            28,
+            29,
+            30,
+            31,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39,
+            40,
+            41,
+            42,
+            43,
+            44,
+            45,
+            46,
+            47,
+            48,
+            49,
+            50
+        ],
         "image_transforms": {
             "enable": false,
             "max_num_transforms": 3,
@@ -80,12 +132,20 @@
                     6
                 ]
             },
-            "observation.images.front": {
+            "observation.images.gripper": {
+                "type": "VISUAL",
+                "shape": [
+                    3,
+                    480,
+                    640
+                ]
+            },
+            "observation.images.top": {
                 "type": "VISUAL",
                 "shape": [
                     3,
-                    720,
-                    1280
+                    480,
+                    640
                 ]
             }
         },
@@ -97,7 +157,7 @@
                 ]
             }
         },
-        "device": "mps",
+        "device": "cuda",
         "use_amp": false,
         "push_to_hub": true,
         "repo_id": "aaron-ser/smolvla-model",
@@ -145,7 +205,7 @@
         "min_period": 0.004,
         "max_period": 4.0
     },
-    "output_dir": "/Users/aaronserpilin/Documents/Storks/storks-so101/my_robot/outputs/train/smolvla_test",
+    "output_dir": "/home/jupyter/storks-so101/my_robot/outputs/outputs/train/smolvla_test",
     "job_name": "smolvla_test",
     "resume": false,
     "seed": 1000,
@@ -186,7 +246,7 @@
         "project": "lerobot",
         "entity": null,
         "notes": null,
-        "run_id": "9t6jp7t3",
+        "run_id": "ghq16lgv",
         "mode": null
     }
 }