|
|
|
@ -64,7 +64,6 @@ class SpecAugmentor(AugmentorBase):
|
|
|
|
|
self.n_freq_masks = n_freq_masks
|
|
|
|
|
self.n_time_masks = n_time_masks
|
|
|
|
|
self.p = p
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# adaptive SpecAugment
|
|
|
|
|
self.adaptive_number_ratio = adaptive_number_ratio
|
|
|
|
@ -120,7 +119,7 @@ class SpecAugmentor(AugmentorBase):
|
|
|
|
|
@property
|
|
|
|
|
def time_mask(self):
|
|
|
|
|
return self._time_mask
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def __repr__(self):
|
|
|
|
|
return f"specaug: F-{F}, T-{T}, F-n-{n_freq_masks}, T-n-{n_time_masks}"
|
|
|
|
|
|
|
|
|
|