提交 d64cdc78 编写于 作者: H Hui Zhang

fix

上级 c484d537
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
## 安装 ## 安装
在以下环境测试验证过: 在以下环境测试验证过:
* Ubuntu 16.04 * Ubuntu 16.04
* python>=3.7 * python>=3.7
......
...@@ -233,7 +233,7 @@ class U2Trainer(Trainer): ...@@ -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): ...@@ -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): ...@@ -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}"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册