multi workers for valid dataloader

pull/891/head
Hui Zhang 3 years ago
parent 3c32a292e9
commit d14e56476f

@ -86,7 +86,8 @@ class U2Trainer(Trainer):
batch_size=config.collator.batch_size,
shuffle=False,
drop_last=False,
collate_fn=collate_fn_dev)
collate_fn=collate_fn_dev,
num_workers=config.collator.num_workers, )
# test dataset, return raw text
config.data.manifest = config.data.test_manifest

@ -265,7 +265,7 @@ class U2Trainer(Trainer):
batch_frames_out=0,
batch_frames_inout=0,
preprocess_conf=None,
n_iter_processes=1,
n_iter_processes=config.collator.num_workers,
subsampling_factor=1,
num_encs=1)

Loading…
Cancel
Save