Upload policy weights, train config and readme

config.json

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

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7276357f94c1281e2b7fe77b9006825bb1bd61dcb48c277621866c4512a06d1c
+oid sha256:7b341b44007fd3b11e5c55f177135dd6c6d92283eb656bd098ea53b19b9e1fd8
 size 346095184

train_config.json

@@ -182,7 +182,7 @@
         "dropout": 0.1,
         "dim_feedforward": 4096,
         "activation": "gelu",
-        "training_noise_sampling": "uniform",
+        "training_noise_sampling": "beta",
         "clip_sample": true,
         "clip_sample_range": 1.0,
         "num_inference_steps": 20,
@@ -197,7 +197,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/train/2026-01-14/21-49-54_pusht_jit",
+    "output_dir": "outputs/train/2026-01-15/12-59-35_pusht_jit",
     "job_name": "pusht_jit",
     "resume": false,
     "seed": 1000,
@@ -237,7 +237,7 @@
         "project": "Jan",
         "entity": null,
         "notes": null,
-        "run_id": "fept79yx",
+        "run_id": "u62brwl4",
         "mode": null
     },
     "use_rabc": false,