diff --git a/README_cn.md b/README_cn.md index 29aadbdf639b529ce049f8244acdba7394a2e3de..4b92736252ef7249e204cecc4ed80349a7316299 100644 --- a/README_cn.md +++ b/README_cn.md @@ -16,7 +16,7 @@ ## 安装 -在以下环境测试验证过: +在以下环境测试验证过: * Ubuntu 16.04 * python>=3.7 diff --git a/deepspeech/exps/u2_kaldi/model.py b/deepspeech/exps/u2_kaldi/model.py index a2f062a18ed4d07074d76663631df9a1d72c958d..60f070a3b32e7252646d54450f490790eed71eb3 100644 --- a/deepspeech/exps/u2_kaldi/model.py +++ b/deepspeech/exps/u2_kaldi/model.py @@ -233,7 +233,7 @@ class U2Trainer(Trainer): batch_frames_inout=0, preprocess_conf=config.collator.augmentation_config, n_iter_processes=config.collator.num_workers, - subsampling_factor=0, + subsampling_factor=1, num_encs=1) self.valid_loader = BatchDataLoader( diff --git a/deepspeech/frontend/augmentor/spec_augment.py b/deepspeech/frontend/augmentor/spec_augment.py index ed593da4ad1dfff5ac0547152f693e2744a449f4..bfa8300af1576cc6c89bb2e3353e6706633ac02a 100644 --- a/deepspeech/frontend/augmentor/spec_augment.py +++ b/deepspeech/frontend/augmentor/spec_augment.py @@ -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}"