fix default dist_samlper to False

pull/1273/head
Hui Zhang 3 years ago
parent 3a2db414e6
commit 45832f6770

@ -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):
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:

Loading…
Cancel
Save