Check number of speakers in dataset matches FS2 params

Author: OscarVanL

examples/fastspeech2_libritts/fastspeech2_dataset.py

@@ -118,12 +118,6 @@ def __init__(
         self.energy_load_fn = energy_load_fn
         self.mel_length_threshold = mel_length_threshold
         self.speakers_map = speakers_map
-        # sp_id = 0
-        # for i in self.utt_ids:
-        #     sp_name = i.split("_")[0]
-        #     if sp_name not in self.speakers_map:
-        #         self.speakers_map[sp_name] = sp_id
-        #         sp_id += 1
         self.speakers = [
             self.speakers_map[i.split("_")[0]] for i in self.utt_ids
         ]  # TODO change but at the moment mfa folder name = speaker name

examples/fastspeech2_libritts/train_fastspeech2.py

@@ -359,8 +359,10 @@ def main():
     with open(args.dataset_mapping) as f:
         dataset_mapping = json.load(f)
         speakers_map = dataset_mapping["speakers_map"]
-        print("SPEAKERS MAP TYPE IS: " , type(speakers_map))
-        print(speakers_map)
+
+    # Check n_speakers matches number of speakers in speakers_map
+    n_speakers = config["fastspeech2_params"]["n_speakers"]
+    assert n_speakers == len(speakers_map)
 
     # define train/valid dataset
     train_dataset = CharactorDurationF0EnergyMelDataset(