wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix in compute embedding without eval partition

Browse Source
This commit is contained in:
Edresson Casanova
2022-04-26 13:58:03 -03:00
parent a41e860a66
commit 6233f4fcd7
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/bin/compute_embeddings.py
+8 -4
View File
@@ -32,15 +32,19 @@ parser.add_argument("output_path", type=str, help="path for output speakers.json
parser.add_argument( parser.add_argument(
"--old_file", type=str, help="Previous speakers.json file, only compute for new audios.", default=None "--old_file", type=str, help="Previous speakers.json file, only compute for new audios.", default=None
) )
parser.add_argument("--use_cuda", type=bool, help="flag to set cuda.", default=True) parser.add_argument("--use_cuda", type=bool, help="flag to set cuda. Default False", default=False)
parser.add_argument("--eval", type=bool, help="compute eval.", default=True) parser.add_argument("--no_eval", type=bool, help="Do not compute eval?. Default False", default=False)
args = parser.parse_args() args = parser.parse_args()
c_dataset = load_config(args.config_dataset_path) c_dataset = load_config(args.config_dataset_path)
meta_data_train, meta_data_eval = load_tts_samples(c_dataset.datasets, eval_split=args.eval) meta_data_train, meta_data_eval = load_tts_samples(c_dataset.datasets, eval_split=not args.no_eval)
wav_files = meta_data_train + meta_data_eval
if meta_data_eval is None:
wav_files = meta_data_train
else:
wav_files = meta_data_train + meta_data_eval
encoder_manager = SpeakerManager( encoder_manager = SpeakerManager(
encoder_model_path=args.model_path, encoder_model_path=args.model_path,