Update hyperparams.yaml

hyperparams.yaml

@@ -63,37 +63,18 @@ use_rand_shift: false
 min_shift: -8000
 max_shift: 8000
 
-# Speed perturbation
-speed_changes: &id001 [95, 100, 105]
-
-# Frequency drop: randomly drops a number of frequency bands to zero.
-speed_perturb: !new:speechbrain.augment.time_domain.SpeedPerturb
-  orig_freq: 48000
-  speeds: *id001
-drop_freq_low: 0  # Min frequency band dropout probability
-drop_freq_high: 1  # Max frequency band dropout probability
-drop_freq_count_low: 1  # Min number of frequency bands to drop
-drop_freq_count_high: 3  # Max number of frequency bands to drop
-drop_freq_width: 0.05  # Width of frequency bands to drop
-
-drop_freq: !new:speechbrain.augment.time_domain.DropFreq
-  drop_freq_low: 0
-  drop_freq_high: 1
-  drop_freq_count_low: 1
-  drop_freq_count_high: 3
-  drop_freq_width: 0.05
-
-# Time drop: randomly drops a number of temporal chunks.
-drop_chunk_count_low: 1  # Min number of audio chunks to drop
-drop_chunk_count_high: 5  # Max number of audio chunks to drop
-drop_chunk_length_low: 1000  # Min length of audio chunks to drop
-drop_chunk_length_high: 2000  # Max length of audio chunks to drop
-
-drop_chunk: !new:speechbrain.augment.time_domain.DropChunk
-  drop_length_low: 1000
-  drop_length_high: 2000
-  drop_count_low: 1
-  drop_count_high: 5
+speedperturb: !new:speechbrain.lobes.augment.TimeDomainSpecAugment
+  perturb_prob: 1.0
+  drop_freq_prob: 0.0
+  drop_chunk_prob: 0.0
+  sample_rate: 8000
+  speeds: [95, 100, 105]
+
+wavedrop: !new:speechbrain.lobes.augment.TimeDomainSpecAugment
+  perturb_prob: 0.0
+  drop_freq_prob: 1.0
+  drop_chunk_prob: 1.0
+  sample_rate: 48000
 
 # loss thresholding -- this thresholds the training loss
 threshold_byloss: true