Upload folder using huggingface_hub

config.json

@@ -13,7 +13,7 @@
                 6
             ]
         },
-        "observation.images.Camera 1": {
+        "observation.images.Camera_1": {
             "type": "VISUAL",
             "shape": [
                 3,
@@ -21,7 +21,7 @@
                 640
             ]
         },
-        "observation.images.Camera 2": {
+        "observation.images.Camera_2": {
             "type": "VISUAL",
             "shape": [
                 3,
@@ -39,7 +39,7 @@
         }
     },
     "device": "cuda",
-    "use_amp": false,
+    "use_amp": true,
     "horizon": 16,
     "n_action_steps": 8,
     "drop_n_last_frames": 7,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:adedb31cce64eb1de11aa88a090578a9b7da4a5c71477a1eed67146d77573cf3
+oid sha256:b53ddf8cbfe50a31fa83668c53f849cbf8ca55e60c71f042bb3fec87b0164824
 size 1066517752

train_config.json

@@ -1,7 +1,7 @@
 {
     "dataset": {
-        "repo_id": "zero7101/so100_Dataset_14",
-        "root": null,
+        "repo_id": "so100_Dataset_1000_V0.1",
+        "root": "so100_Dataset_1000_V0.1",
         "episodes": null,
         "image_transforms": {
             "enable": false,
@@ -80,7 +80,7 @@
                     6
                 ]
             },
-            "observation.images.Camera 1": {
+            "observation.images.Camera_1": {
                 "type": "VISUAL",
                 "shape": [
                     3,
@@ -88,7 +88,7 @@
                     640
                 ]
             },
-            "observation.images.Camera 2": {
+            "observation.images.Camera_2": {
                 "type": "VISUAL",
                 "shape": [
                     3,
@@ -106,7 +106,7 @@
             }
         },
         "device": "cuda",
-        "use_amp": false,
+        "use_amp": true,
         "horizon": 16,
         "n_action_steps": 8,
         "drop_n_last_frames": 7,
@@ -149,17 +149,17 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/train/2025-06-12/17-45-21_diffusion",
-    "job_name": "diffusion",
+    "output_dir": "outputs/train/my_diffusion_fast_2",
+    "job_name": "my_diffusion_training_fast",
     "resume": false,
     "seed": 1000,
-    "num_workers": 4,
-    "batch_size": 128,
+    "num_workers": 6,
+    "batch_size": 256,
     "steps": 100000,
     "eval_freq": 20000,
     "log_freq": 200,
     "save_checkpoint": true,
-    "save_freq": 20000,
+    "save_freq": 500,
     "use_policy_training_preset": true,
     "optimizer": {
         "type": "adam",
@@ -183,7 +183,7 @@
         "use_async_envs": false
     },
     "wandb": {
-        "enable": false,
+        "enable": true,
         "disable_artifact": false,
         "project": "lerobot",
         "entity": null,