Reformatting

This commit is contained in:
Piotr Żelasko 2021-10-09 14:45:46 +00:00
parent b682467e4d
commit 069ebaf9ba
2 changed files with 8 additions and 2 deletions

View File

@ -305,7 +305,10 @@ class LibriSpeechAsrDataModule(DataModule):
)
logging.debug("About to create test dataloader")
test_dl = DataLoader(
test, batch_size=None, sampler=sampler, num_workers=self.args.num_workers
test,
batch_size=None,
sampler=sampler,
num_workers=self.args.num_workers,
)
test_loaders.append(test_dl)

View File

@ -230,7 +230,10 @@ class YesNoAsrDataModule(DataModule):
)
logging.debug("About to create test dataloader")
test_dl = DataLoader(
test, batch_size=None, sampler=sampler, num_workers=self.args.num_workers
test,
batch_size=None,
sampler=sampler,
num_workers=self.args.num_workers,
)
return test_dl