From d64cdc7838468771e4037ca2cfaa6f1089a1ee80 Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Thu, 19 Aug 2021 04:40:06 +0000 Subject: [PATCH] fix --- README_cn.md | 2 +- deepspeech/exps/u2_kaldi/model.py | 2 +- deepspeech/frontend/augmentor/spec_augment.py | 3 +-- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/README_cn.md b/README_cn.md index 29aadbdf..4b927362 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 a2f062a1..60f070a3 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 ed593da4..bfa8300a 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}" -- GitLab