Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Ignore NaN embeddings on the compute embeddings script

This commit is contained in:
Edresson Casanova 2022-07-04 14:56:41 -03:00
parent f9199b04c4
commit 5e6bd601d0
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TTS/bin/compute_embeddings.py
View File

@ -1,5 +1,6 @@
import argparse import argparse
import os import os
import numpy as np
from argparse import RawTextHelpFormatter from argparse import RawTextHelpFormatter
import torch import torch
@ -68,6 +69,11 @@ for idx, wav_file in enumerate(tqdm(wav_files)):
# extract the embedding # extract the embedding
embedd = encoder_manager.compute_embedding_from_clip(wav_file) embedd = encoder_manager.compute_embedding_from_clip(wav_file)
# ignore samples that the embedding is NaN and return a warning
if np.isnan(np.array(embedd)).any():
print(f"> The embedding of the file {wav_file} was not saved, because it has NaN values, It probably means that this file is empty, please check it!")
continue
if args.use_predicted_label: if args.use_predicted_label:
map_classid_to_classname = getattr(encoder_manager.encoder_config, "map_classid_to_classname", None) map_classid_to_classname = getattr(encoder_manager.encoder_config, "map_classid_to_classname", None)
if encoder_manager.encoder_criterion is not None and map_classid_to_classname is not None: if encoder_manager.encoder_criterion is not None and map_classid_to_classname is not None: