reduce time cutoff

EleutherAI · Apr 24, 2024 · 8154b5d · 8154b5d
1 parent ada0673
commit 8154b5d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/amt/data.py b/amt/data.py
@@ -88,9 +88,9 @@ def get_wav_mid_segments(
                 max_pedal_len_ms=15000,
             )
 
-            # Hardcoded to 10s
-            if _check_onset_threshold(mid_feature, 10500) is False:
-                print("No note messages after 10s - skipping")
+            # Hardcoded to 5s
+            if _check_onset_threshold(mid_feature, 5000) is False:
+                print("No note messages after 5s - skipping")
                 continue
 
         else: