Upload policy weights, train config and readme

README.md

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

config.json

@@ -60,9 +60,9 @@
     "use_group_norm": true,
     "spatial_softmax_num_keypoints": 32,
     "use_separate_rgb_encoder_per_camera": true,
-    "audio_backbone": "ramen-noodels/red_round_button_small_click_detector",
-    "audio_norm_mean": 0.6464800529520275,
-    "audio_norm_std": 0.0613050481128137,
+    "audio_backbone": "ramen-noodels/red_round_button_small_click_detector_n25",
+    "audio_norm_mean": 0.6462260507515563,
+    "audio_norm_std": 0.06125792184123532,
     "audio_feature_type": "embedding",
     "time_dimension": 298,
     "freeze_audio_encoder": false,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:47ffef4c80e88a498d33f7272fcdaa5cb652105eaefd8f7b3ec1545762463d1c
+oid sha256:648d9d98c8ac49694bf926822a34026ae2dbd0a7bfeb029f3071cede2cbd2c9d
 size 1672743388

train_config.json

@@ -128,9 +128,9 @@
         "use_group_norm": true,
         "spatial_softmax_num_keypoints": 32,
         "use_separate_rgb_encoder_per_camera": true,
-        "audio_backbone": "ramen-noodels/red_round_button_small_click_detector",
-        "audio_norm_mean": 0.6464800529520275,
-        "audio_norm_std": 0.0613050481128137,
+        "audio_backbone": "ramen-noodels/red_round_button_small_click_detector_n25",
+        "audio_norm_mean": 0.6462260507515563,
+        "audio_norm_std": 0.06125792184123532,
         "audio_feature_type": "embedding",
         "time_dimension": 298,
         "freeze_audio_encoder": false,
@@ -165,7 +165,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/train/2026-05-30/09-05-14_red_round_button_small_n25_button",
+    "output_dir": "outputs/train/2026-06-12/04-35-23_red_round_button_small_n25_button",
     "job_name": "red_round_button_small_n25_button",
     "resume": false,
     "seed": 2025,
@@ -205,7 +205,7 @@
         "project": "imitation_button",
         "entity": "rtalwar",
         "notes": null,
-        "run_id": "ekovp7rz",
+        "run_id": "vfamob4h",
         "mode": null
     },
     "checkpoint_path": null,