mirror of
https://github.com/coqui-ai/TTS.git
synced 2026-02-24 12:10:16 +01:00
drop last = true
This commit is contained in:
2
train.py
2
train.py
@@ -333,7 +333,7 @@ def main(args):
|
||||
|
||||
train_loader = DataLoader(train_dataset, batch_size=c.batch_size,
|
||||
shuffle=False, collate_fn=train_dataset.collate_fn,
|
||||
drop_last=False, num_workers=c.num_loader_workers,
|
||||
drop_last=True, num_workers=c.num_loader_workers,
|
||||
pin_memory=True)
|
||||
|
||||
val_dataset = Dataset(os.path.join(c.data_path, c.meta_file_val),
|
||||
|
||||
Reference in New Issue
Block a user