max_num_masks is at least min_num_masks (#37)

rwth-i6 · Oct 12, 2023 · 48ade42 · 48ade42
1 parent aaa51af
commit 48ade42
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/i6_models/primitives/specaugment.py b/i6_models/primitives/specaugment.py
@@ -1,3 +1,4 @@
+import numpy as np
 import torch
 
 
@@ -129,7 +130,7 @@ def specaugment_v1_by_length(
     return specaugment_v1(
         audio_features,
         time_min_num_masks=time_min_num_masks,
-        time_max_num_masks=audio_features.size(1) // time_max_mask_per_n_frames,
+        time_max_num_masks=np.maximum(audio_features.size(1) // time_max_mask_per_n_frames, time_min_num_masks),
         time_mask_max_size=time_mask_max_size,
         freq_min_num_masks=freq_min_num_masks,
         freq_max_num_masks=freq_max_num_masks,