Upload folder using huggingface_hub

config.json

@@ -10,15 +10,15 @@
         "observation.state": {
             "type": "STATE",
             "shape": [
-                12
+                6
             ]
         },
         "observation.image.front": {
             "type": "VISUAL",
             "shape": [
                 3,
-                128,
-                128
+                480,
+                640
             ]
         }
     },
@@ -30,11 +30,11 @@
             ]
         }
     },
-    "device": "cuda",
+    "device": "mps",
     "use_amp": false,
     "use_peft": false,
-    "push_to_hub": true,
-    "repo_id": "orellius/so101_matchbox_fpv_act",
+    "push_to_hub": false,
+    "repo_id": null,
     "private": null,
     "tags": null,
     "license": null,
@@ -56,7 +56,7 @@
     "temporal_ensemble_coeff": null,
     "dropout": 0.1,
     "kl_weight": 10.0,
-    "optimizer_lr": 5e-05,
+    "optimizer_lr": 0.0001,
     "optimizer_weight_decay": 0.0001,
     "optimizer_lr_backbone": 1e-05
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:28d2684830bbc779797471afd5364a2eabb00b23c1034fff74eeb7965235b449
-size 206725424
+oid sha256:8958af67344b1515d924e54906f04f7cb7c3420e6347322a07b71dc42f1b594c
+size 206700792

train_config.json

@@ -1,6 +1,6 @@
 {
     "dataset": {
-        "repo_id": "hubnemo/so101_matchbox_reward",
+        "repo_id": "hubnemo/so101_matchbox_reward_dataset_fpv",
         "root": null,
         "episodes": null,
         "image_transforms": {
@@ -77,15 +77,15 @@
             "observation.state": {
                 "type": "STATE",
                 "shape": [
-                    12
+                    6
                 ]
             },
             "observation.image.front": {
                 "type": "VISUAL",
                 "shape": [
                     3,
-                    128,
-                    128
+                    480,
+                    640
                 ]
             }
         },
@@ -97,11 +97,11 @@
                 ]
             }
         },
-        "device": "cuda",
+        "device": "mps",
         "use_amp": false,
         "use_peft": false,
-        "push_to_hub": true,
-        "repo_id": "orellius/so101_matchbox_fpv_act",
+        "push_to_hub": false,
+        "repo_id": null,
         "private": null,
         "tags": null,
         "license": null,
@@ -123,12 +123,12 @@
         "temporal_ensemble_coeff": null,
         "dropout": 0.1,
         "kl_weight": 10.0,
-        "optimizer_lr": 5e-05,
+        "optimizer_lr": 0.0001,
         "optimizer_weight_decay": 0.0001,
         "optimizer_lr_backbone": 1e-05
     },
-    "output_dir": "outputs/train/matchbox_so101_matchbox_act_20250822_1917",
-    "job_name": "matchbox",
+    "output_dir": "outputs/train/matchbox_so101_matchbox_act_20250905_1435",
+    "job_name": "matchbox_so101_matchbox_act_20250905_1435",
     "resume": false,
     "seed": 1000,
     "num_workers": 4,
@@ -139,13 +139,13 @@
     "save_checkpoint": true,
     "save_freq": 20000,
     "use_validation": true,
-    "val_split": 0.05,
-    "val_freq": 100,
+    "val_split": 0.02,
+    "val_freq": 50,
     "val_batch_size": 32,
     "use_policy_training_preset": true,
     "optimizer": {
         "type": "adamw",
-        "lr": 5e-05,
+        "lr": 0.0001,
         "weight_decay": 0.0001,
         "grad_clip_norm": 10.0,
         "betas": [
@@ -166,7 +166,7 @@
         "project": "lerobot",
         "entity": null,
         "notes": null,
-        "run_id": "z2x1oh3b",
+        "run_id": "zsnktxth",
         "mode": null
     },
     "use_peft": false,