提交 45832f67 编写于 作者: H Hui Zhang

fix default dist_samlper to False

上级 3a2db414
...@@ -241,7 +241,7 @@ class U2Trainer(Trainer): ...@@ -241,7 +241,7 @@ class U2Trainer(Trainer):
n_iter_processes=config.num_workers, n_iter_processes=config.num_workers,
subsampling_factor=1, subsampling_factor=1,
num_encs=1, num_encs=1,
dist_sampler=True, dist_sampler=False,
shortest_first=False) shortest_first=False)
self.valid_loader = BatchDataLoader( self.valid_loader = BatchDataLoader(
...@@ -262,7 +262,7 @@ class U2Trainer(Trainer): ...@@ -262,7 +262,7 @@ class U2Trainer(Trainer):
n_iter_processes=config.num_workers, n_iter_processes=config.num_workers,
subsampling_factor=1, subsampling_factor=1,
num_encs=1, num_encs=1,
dist_sampler=True, dist_sampler=False,
shortest_first=False) shortest_first=False)
logger.info("Setup train/valid Dataloader!") logger.info("Setup train/valid Dataloader!")
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册