Mirrors/voice-cloning-collab
1
0
Fork 0
mirror of https://github.com/liuhaozhe6788/voice-cloning-collab.git synced 2026-05-18 05:04:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

在dataloader中增加shuffle

Browse Source
This commit is contained in:
liuhaozhe6788
2022-11-17 20:50:13 +08:00
parent 925afce75d
commit d03314f336
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
encoder/data_objects/speaker_verification_dataset.py
View File

@@ -53,7 +53,7 @@ class Dev_Dataset(Dataset):
class DataLoader(DataLoader):
def __init__(self, dataset, speakers_per_batch, utterances_per_speaker, sampler=None,
def __init__(self, dataset, speakers_per_batch, utterances_per_speaker, shuffle, sampler=None,
batch_sampler=None, num_workers=0, pin_memory=False, timeout=0,
worker_init_fn=None):
self.utterances_per_speaker = utterances_per_speaker
@@ -61,7 +61,7 @@ class DataLoader(DataLoader):
super().__init__(
dataset=dataset,
batch_size=speakers_per_batch,
shuffle=False,
shuffle=shuffle,
sampler=sampler,
batch_sampler=batch_sampler,
num_workers=num_workers,