pull/1091/head
Hui Zhang 4 years ago
parent dce2565a97
commit 9527f547d0

@ -16,7 +16,7 @@
## 安装 ## 安装
在以下环境测试验证过: 在以下环境测试验证过:
* Ubuntu 16.04 * Ubuntu 16.04
* python>=3.7 * python>=3.7

@ -233,7 +233,7 @@ class U2Trainer(Trainer):
batch_frames_inout=0, batch_frames_inout=0,
preprocess_conf=config.collator.augmentation_config, preprocess_conf=config.collator.augmentation_config,
n_iter_processes=config.collator.num_workers, n_iter_processes=config.collator.num_workers,
subsampling_factor=0, subsampling_factor=1,
num_encs=1) num_encs=1)
self.valid_loader = BatchDataLoader( self.valid_loader = BatchDataLoader(

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

Loading…
Cancel
Save