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

Merge pull request #407 from thllwg/dev 

Configurable number of workers for DataLoader
This commit is contained in:
Eren Gölge 2020-05-16 15:45:07 +02:00 committed by GitHub
parent 93cab61d80 65b9c7d3d6
commit d023872f0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
speaker_encoder/config.json
View File

@ -34,6 +34,7 @@
"save_step": 1000, // Number of training steps expected to save traning stats and checkpoints. "save_step": 1000, // Number of training steps expected to save traning stats and checkpoints.
"print_step": 1, // Number of steps to log traning on console. "print_step": 1, // Number of steps to log traning on console.
"output_path": "/media/erogol/data_ssd/Models/libri_tts/speaker_encoder/", // DATASET-RELATED: output path for all training outputs. "output_path": "/media/erogol/data_ssd/Models/libri_tts/speaker_encoder/", // DATASET-RELATED: output path for all training outputs.
"num_loader_workers": 0, // number of training data loader processes. Don't set it too big. 4-8 are good values.
"model": { "model": {
"input_dim": 40, "input_dim": 40,
"proj_dim": 128, "proj_dim": 128,

2
speaker_encoder/train.py
View File

@ -44,7 +44,7 @@ def setup_loader(ap, is_val=False, verbose=False):
loader = DataLoader(dataset, loader = DataLoader(dataset,
batch_size=c.num_speakers_in_batch, batch_size=c.num_speakers_in_batch,
shuffle=False, shuffle=False,
num_workers=0, num_workers=c.num_loader_workers,
collate_fn=dataset.collate_fn) collate_fn=dataset.collate_fn)
return loader return loader