-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathdemo_config.py
53 lines (45 loc) · 3.13 KB
/
demo_config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from pathlib import Path
#------------------------------------------------------------------------------------
gpu_id = '0,1,2,3,4,5,6,7'
#------------------------------------------------------------------------------------
# DeepTalk encoder (model_GST: fCNN -> GST) trained on Librispeech, VoxCeleb1, and VoxCeleb2 Data
# Generic Model
# enc_model_fpath = Path('trained_models/Generic/Encoder/model_GST.pt')
# enc_module_name = "model_GST"
# syn_model_dir = Path('trained_models/Generic/Synthesizer/logs-model_GST/taco_pretrained/')
# voc_model_fpath = Path('trained_models/Generic/Vocoder/model_GST/model_GST.pt')
#------------------------------------------------------------------------------------
# DeepTalk encoder (model_GST: fCNN -> GST) trained on Librispeech, VoxCeleb1, and VoxCeleb2 Data
# and finetuned on a specific male speaker
# enc_model_fpath = Path('trained_models/Sample_Male/Encoder/model_GST.pt')
# enc_module_name = "model_GST"
# syn_model_dir = Path('trained_models/Sample_Male/Synthesizer/logs-model_GST_ft/taco_pretrained/')
# voc_model_fpath = Path('trained_models/Sample_Male/Vocoder/model_GST_ft/model_GST_ft.pt')
#------------------------------------------------------------------------------------
# DeepTalk encoder (model_GST: fCNN -> GST) trained on Librispeech, VoxCeleb1, and VoxCeleb2 Data
# and finetuned on a specific female speaker
# enc_model_fpath = Path('trained_models/Sample_Female/Encoder/model_GST.pt')
# enc_module_name = "model_GST"
# syn_model_dir = Path('trained_models/Sample_Female/Synthesizer/logs-model_GST_ft/taco_pretrained/')
# voc_model_fpath = Path('trained_models/Sample_Female/Vocoder/model_GST_ft/model_GST_ft.pt')
#------------------------------------------------------------------------------------
# DeepTalk encoder (model_GST: fCNN -> GST) trained on Librispeech, VoxCeleb1, and VoxCeleb2 Data
# and finetuned on Hannah
# enc_model_fpath = Path('trained_models/Hannah/Encoder/model_GST.pt')
# enc_module_name = "model_GST"
# syn_model_dir = Path('trained_models/Hannah/Synthesizer/logs-model_GST_ft/taco_pretrained/')
# voc_model_fpath = Path('trained_models/Hannah/Vocoder/model_GST_ft/model_GST_ft.pt')
#------------------------------------------------------------------------------------
# DeepTalk encoder (model_GST: fCNN -> GST) trained on Librispeech, VoxCeleb1, and VoxCeleb2 Data
# and finetuned on Ted
# enc_model_fpath = Path('trained_models/Ted/Encoder/model_GST.pt')
# enc_module_name = "model_GST"
# syn_model_dir = Path('trained_models/Ted/Synthesizer/logs-model_GST_ft/taco_pretrained/')
# voc_model_fpath = Path('trained_models/Ted/Vocoder/model_GST_ft/model_GST_ft.pt')
#------------------------------------------------------------------------------------
# DeepTalk encoder (model_GST: fCNN -> GST) trained on Librispeech, VoxCeleb1, and VoxCeleb2 Data
# and finetuned on Obama
enc_model_fpath = Path('trained_models/Obama/Encoder/model_GST.pt')
enc_module_name = "model_GST"
syn_model_dir = Path('trained_models/Obama/Synthesizer/logs-model_GST_ft/taco_pretrained/')
voc_model_fpath = Path('trained_models/Obama/Vocoder/model_GST_ft/model_GST_ft.pt')