fixes (dataset length + coquille)

Author: NathanGodey

data_loader_circular.py

@@ -38,7 +38,7 @@ def __init__(self, root_dir, len_crop):
         self.train_dataset = list(dataset)
         self.num_tokens = len(self.train_dataset)
 
-        print('Finished loading the dataset...')
+        print(f'Finished loading the dataset...')
 
 
     def load_data(self, submeta, dataset, idx_offset):
@@ -75,8 +75,8 @@ def __getitem__(self, index):
         return uttr, emb_org, emb_trgt
 
     def __len__(self):
-        """Return the number of spkrs."""
-        return self.num_tokens
+        """Return the number of spkrs combinations."""
+        return (self.num_tokens * (self.num_tokens - 1))
 
 
 

make_metadata.py

@@ -26,7 +26,10 @@ def load_speaker_embedding_model():
 
 def make_metadata(dataset_dir = 'training_set'):
 
-    C=load_speaker_embedding_model().eval()
+    num_uttrs = 10
+    len_crop = 128
+
+    C = load_speaker_embedding_model().eval()
 
     # Directory containing mel-spectrograms
     rootDir = dataset_dir + '/spmel'