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

multi workers for valid dataloader

上级 3c32a292
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册