Upload policy weights, train config and readme

README.md

@@ -5,8 +5,8 @@ license: apache-2.0
 model_name: diffusion
 pipeline_tag: robotics
 tags:
-- lerobot
 - robotics
+- lerobot
 - diffusion
 ---
 

config.json

@@ -17,7 +17,7 @@
             ]
         }
     },
-    "device": "cpu",
+    "device": "cuda",
     "use_amp": false,
     "use_peft": false,
     "push_to_hub": true,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c84f0041aa02320a1b57ac443775d9406519ed7b4711c05ceecc4ea7d69b9fbc
+oid sha256:586c65b8b955d8bfd468296f474e64da9df3d02f7acfb410b76cfd4d4ceec269
 size 992360344

train_config.json

@@ -100,7 +100,7 @@
                 ]
             }
         },
-        "device": "cpu",
+        "device": "cuda",
         "use_amp": false,
         "use_peft": false,
         "push_to_hub": true,
@@ -158,7 +158,7 @@
         "scheduler_warmup_steps": 500
     },
     "reward_model": null,
-    "output_dir": "outputs\\train\\2026-05-14\\21-50-14_diffusion",
+    "output_dir": "outputs\\train\\2026-05-14\\22-15-30_diffusion",
     "job_name": "diffusion",
     "resume": false,
     "seed": 1000,
@@ -167,7 +167,7 @@
     "batch_size": 16,
     "prefetch_factor": 4,
     "persistent_workers": true,
-    "steps": 10,
+    "steps": 5000,
     "eval_freq": 20000,
     "log_freq": 200,
     "tolerance_s": 0.0001,