@ -15,6 +15,7 @@
## Setup
All tested under:
* Ubuntu 16.04
* python>=3.7
* paddlepaddle>=2.1.2
@ -16,6 +16,8 @@
## 安装
在以下环境测试验证过:
@ -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=1,
subsampling_factor=0,
num_encs=1)
self.valid_loader = BatchDataLoader(
@ -55,6 +55,8 @@ class CustomConverter():
xs = [x[::self.subsampling_factor, :] for x in xs]
# get batch of lengths of input sequences
print(xs)
print(ys)
ilens = np.array([x.shape[0] for x in xs])
# perform padding and convert to tensor