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
 - jit
 - lerobot
-- robotics
 ---
 
 # Model Card for jit

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": 100,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a0b840775799a3b8447df4c63cab612405dafd7cf702f322668e2634d3e46180
-size 341370208
+oid sha256:52e646f6e7bc2bf165657cfd8264b75c01b46128e62485192ad4219faf7dfaec
+size 260841784

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": 100,
@@ -197,18 +197,18 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/train/2026-01-10/18-46-38_pusht_jit",
+    "output_dir": "outputs/train/2026-01-22/20-09-50_pusht_jit",
     "job_name": "pusht_jit",
     "resume": false,
-    "seed": 1000,
+    "seed": 10000,
     "num_workers": 4,
     "batch_size": 64,
     "steps": 200000,
-    "eval_freq": 20000,
+    "eval_freq": 25000,
     "log_freq": 200,
     "tolerance_s": 0.0001,
     "save_checkpoint": true,
-    "save_freq": 20000,
+    "save_freq": 25000,
     "use_policy_training_preset": true,
     "optimizer": {
         "type": "adam",
@@ -234,10 +234,10 @@
     "wandb": {
         "enable": true,
         "disable_artifact": false,
-        "project": "Jan",
+        "project": "incontext",
         "entity": null,
         "notes": null,
-        "run_id": "mrjohnkq",
+        "run_id": "2flhsfw9",
         "mode": null
     },
     "use_rabc": false,