This commit is contained in:
mrq 2023-08-14 10:50:40 -05:00
parent 13571380be
commit 5fa86182b5
2 changed files with 2 additions and 5 deletions

View File

@ -99,7 +99,4 @@ inference:
use_vocos: True
bitsandbytes:
enabled: false
device: cuda
distributed: False
enabled: false

View File

@ -313,7 +313,7 @@ def _create_dataloader(dataset, training):
return DataLoader(
dataset=dataset,
batch_size=cfg.hyperparameters.batch_size if training else cfg.evaluation.batch_size,
shuffle=False if cfg.distributed else True, # training
shuffle=False, # if cfg.distributed else True, # training
drop_last=training,
num_workers=cfg.dataset.workers,
collate_fn=collate_fn,