Upload policy weights, train config and readme

README.md

@@ -5,9 +5,9 @@ license: apache-2.0
 model_name: jit
 pipeline_tag: robotics
 tags:
+- robotics
 - lerobot
 - jit
-- robotics
 ---
 
 # Model Card for jit

config.json

@@ -59,7 +59,7 @@
     "dropout": 0.1,
     "dim_feedforward": 4096,
     "activation": "gelu",
-    "training_noise_sampling": "beta",
+    "training_noise_sampling": "uniform",
     "clip_sample": true,
     "clip_sample_range": 1.0,
     "num_inference_steps": 100,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5772b8474b589ea8d9bb08996020eef2364bf75b62012db2b513980cd53b12d5
-size 346095184
+oid sha256:0fdb56e3b0bb725d006274d053e6778ba9c4a7809e88427c4f3d4ac57bc20e4e
+size 261038296

train_config.json

@@ -182,7 +182,7 @@
         "dropout": 0.1,
         "dim_feedforward": 4096,
         "activation": "gelu",
-        "training_noise_sampling": "beta",
+        "training_noise_sampling": "uniform",
         "clip_sample": true,
         "clip_sample_range": 1.0,
         "num_inference_steps": 100,
@@ -197,7 +197,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/train/2026-01-11/16-45-10_pusht_jit",
+    "output_dir": "outputs/train/2026-01-11/20-05-42_pusht_jit",
     "job_name": "pusht_jit",
     "resume": false,
     "seed": 1000,
@@ -237,7 +237,7 @@
         "project": "Jan",
         "entity": null,
         "notes": null,
-        "run_id": "wk7crhru",
+        "run_id": "fr9fe7jd",
         "mode": null
     },
     "use_rabc": false,