pull/768/head
Hui Zhang 3 years ago
parent c09b0e8940
commit c81743403a

@ -15,6 +15,7 @@
## Setup ## Setup
All tested under:
* Ubuntu 16.04 * Ubuntu 16.04
* python>=3.7 * python>=3.7
* paddlepaddle>=2.1.2 * paddlepaddle>=2.1.2

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

@ -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=1, subsampling_factor=0,
num_encs=1) num_encs=1)
self.valid_loader = BatchDataLoader( self.valid_loader = BatchDataLoader(

@ -55,6 +55,8 @@ class CustomConverter():
xs = [x[::self.subsampling_factor, :] for x in xs] xs = [x[::self.subsampling_factor, :] for x in xs]
# get batch of lengths of input sequences # get batch of lengths of input sequences
print(xs)
print(ys)
ilens = np.array([x.shape[0] for x in xs]) ilens = np.array([x.shape[0] for x in xs])
# perform padding and convert to tensor # perform padding and convert to tensor

Loading…
Cancel
Save