Upload policy weights, train config and readme

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_n75",
+    "audio_norm_mean": 0.6455835084383504,
+    "audio_norm_std": 0.06144516918406618,
     "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:d23a725752ff8b6751baa862a6f03a47d3716ebea68ab8e510b256fa0c43cc64
+oid sha256:74b7337ba3cb587ec971bca2ff182dcffd0888a05d10cbcdf2ebed5e23d92015
 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_n75",
+        "audio_norm_mean": 0.6455835084383504,
+        "audio_norm_std": 0.06144516918406618,
         "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-06-09/14-41-44_red_round_button_small_n75_button",
+    "output_dir": "outputs/train/2026-06-12/13-33-14_red_round_button_small_n75_button",
     "job_name": "red_round_button_small_n75_button",
     "resume": false,
     "seed": 2025,
@@ -205,7 +205,7 @@
         "project": "imitation_button",
         "entity": "rtalwar",
         "notes": null,
-        "run_id": "ifp7remp",
+        "run_id": "1aqshocd",
         "mode": null
     },
     "checkpoint_path": null,