ayousanz commited on
Commit
e55f565
·
verified ·
1 Parent(s): 1f46534

Upload folder using huggingface_hub

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1.log +874 -0
  2. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/config.yaml +386 -0
  3. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/batch_keys +2 -0
  4. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/energy_lengths_stats.npz +3 -0
  5. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/energy_stats.npz +3 -0
  6. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/feats_lengths_stats.npz +3 -0
  7. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/feats_stats.npz +3 -0
  8. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/pitch_lengths_stats.npz +3 -0
  9. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/pitch_stats.npz +3 -0
  10. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/speech_shape +13 -0
  11. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/stats_keys +6 -0
  12. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/text_shape +13 -0
  13. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/batch_keys +2 -0
  14. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/energy_lengths_stats.npz +3 -0
  15. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/energy_stats.npz +3 -0
  16. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/feats_lengths_stats.npz +3 -0
  17. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/feats_stats.npz +3 -0
  18. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/pitch_lengths_stats.npz +3 -0
  19. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/pitch_stats.npz +3 -0
  20. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/speech_shape +2 -0
  21. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/stats_keys +6 -0
  22. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/text_shape +2 -0
  23. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2.log +874 -0
  24. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/config.yaml +386 -0
  25. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/batch_keys +2 -0
  26. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/energy_lengths_stats.npz +3 -0
  27. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/energy_stats.npz +3 -0
  28. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/feats_lengths_stats.npz +3 -0
  29. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/feats_stats.npz +3 -0
  30. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/pitch_lengths_stats.npz +3 -0
  31. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/pitch_stats.npz +3 -0
  32. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/speech_shape +13 -0
  33. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/stats_keys +6 -0
  34. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/text_shape +13 -0
  35. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/batch_keys +2 -0
  36. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/energy_lengths_stats.npz +3 -0
  37. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/energy_stats.npz +3 -0
  38. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/feats_lengths_stats.npz +3 -0
  39. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/feats_stats.npz +3 -0
  40. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/pitch_lengths_stats.npz +3 -0
  41. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/pitch_stats.npz +3 -0
  42. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/speech_shape +2 -0
  43. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/stats_keys +6 -0
  44. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/text_shape +2 -0
  45. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3.log +874 -0
  46. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/config.yaml +386 -0
  47. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/train/batch_keys +2 -0
  48. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/train/energy_lengths_stats.npz +3 -0
  49. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/train/energy_stats.npz +3 -0
  50. exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/train/feats_lengths_stats.npz +3 -0
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1.log ADDED
@@ -0,0 +1,874 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # python3 -m espnet2.bin.gan_tts_train --collect_stats true --write_collected_feats false --use_preprocessor true --token_type phn --token_list dump/token_list/phn_jaconv_pyopenjtalk/tokens.txt --non_linguistic_symbols none --cleaner jaconv --g2p pyopenjtalk --normalize none --pitch_normalize none --energy_normalize none --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/text,text,text --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/wav.scp,speech,sound --valid_data_path_and_name_and_type dump/raw/jvs010_dev/text,text,text --valid_data_path_and_name_and_type dump/raw/jvs010_dev/wav.scp,speech,sound --train_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.1.scp --valid_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.1.scp --output_dir exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1 --config conf/tuning/train_jets.yaml --feats_extract fbank --feats_extract_conf n_fft=2048 --feats_extract_conf hop_length=300 --feats_extract_conf win_length=1200 --feats_extract_conf fs=24000 --feats_extract_conf fmin=80 --feats_extract_conf fmax=7600 --feats_extract_conf n_mels=80 --pitch_extract_conf fs=24000 --pitch_extract_conf n_fft=2048 --pitch_extract_conf hop_length=300 --pitch_extract_conf f0max=400 --pitch_extract_conf f0min=80 --energy_extract_conf fs=24000 --energy_extract_conf n_fft=2048 --energy_extract_conf hop_length=300 --energy_extract_conf win_length=1200
2
+ # Started at Tue Mar 4 21:23:26 JST 2025
3
+ #
4
+ /usr/lib/python3/dist-packages/requests/__init__.py:89: RequestsDependencyWarning: urllib3 (2.2.3) or chardet (3.0.4) doesn't match a supported version!
5
+ warnings.warn("urllib3 ({}) or chardet ({}) doesn't match a supported "
6
+ /usr/bin/python3 /work/espnet/espnet2/bin/gan_tts_train.py --collect_stats true --write_collected_feats false --use_preprocessor true --token_type phn --token_list dump/token_list/phn_jaconv_pyopenjtalk/tokens.txt --non_linguistic_symbols none --cleaner jaconv --g2p pyopenjtalk --normalize none --pitch_normalize none --energy_normalize none --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/text,text,text --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/wav.scp,speech,sound --valid_data_path_and_name_and_type dump/raw/jvs010_dev/text,text,text --valid_data_path_and_name_and_type dump/raw/jvs010_dev/wav.scp,speech,sound --train_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.1.scp --valid_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.1.scp --output_dir exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1 --config conf/tuning/train_jets.yaml --feats_extract fbank --feats_extract_conf n_fft=2048 --feats_extract_conf hop_length=300 --feats_extract_conf win_length=1200 --feats_extract_conf fs=24000 --feats_extract_conf fmin=80 --feats_extract_conf fmax=7600 --feats_extract_conf n_mels=80 --pitch_extract_conf fs=24000 --pitch_extract_conf n_fft=2048 --pitch_extract_conf hop_length=300 --pitch_extract_conf f0max=400 --pitch_extract_conf f0min=80 --energy_extract_conf fs=24000 --energy_extract_conf n_fft=2048 --energy_extract_conf hop_length=300 --energy_extract_conf win_length=1200
7
+ [92b100c97f43] 2025-03-04 21:23:29,215 (gan_tts:304) INFO: Vocabulary size: 41
8
+ [92b100c97f43] 2025-03-04 21:23:29,440 (encoder:172) INFO: encoder self-attention layer type = self-attention
9
+ [92b100c97f43] 2025-03-04 21:23:29,563 (encoder:172) INFO: encoder self-attention layer type = self-attention
10
+ [92b100c97f43] 2025-03-04 21:23:31,676 (abs_task:1157) INFO: pytorch.version=1.10.1+cu113, cuda.available=True, cudnn.version=8200, cudnn.benchmark=False, cudnn.deterministic=False
11
+ [92b100c97f43] 2025-03-04 21:23:31,686 (abs_task:1158) INFO: Model structure:
12
+ ESPnetGANTTSModel(
13
+ (feats_extract): LogMelFbank(
14
+ (stft): Stft(n_fft=2048, win_length=1200, hop_length=300, center=True, normalized=False, onesided=True)
15
+ (logmel): LogMel(sr=24000, n_fft=2048, n_mels=80, fmin=80, fmax=7600, htk=False)
16
+ )
17
+ (pitch_extract): Dio()
18
+ (energy_extract): Energy(
19
+ (stft): Stft(n_fft=2048, win_length=1200, hop_length=300, center=True, normalized=False, onesided=True)
20
+ )
21
+ (tts): JETS(
22
+ (generator): JETSGenerator(
23
+ (encoder): Encoder(
24
+ (embed): Sequential(
25
+ (0): Embedding(41, 256, padding_idx=0)
26
+ (1): ScaledPositionalEncoding(
27
+ (dropout): Dropout(p=0.2, inplace=False)
28
+ )
29
+ )
30
+ (encoders): MultiSequential(
31
+ (0): EncoderLayer(
32
+ (self_attn): MultiHeadedAttention(
33
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
34
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
35
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
36
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
37
+ (dropout): Dropout(p=0.2, inplace=False)
38
+ )
39
+ (feed_forward): MultiLayeredConv1d(
40
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
41
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
42
+ (dropout): Dropout(p=0.2, inplace=False)
43
+ )
44
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
45
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
46
+ (dropout): Dropout(p=0.2, inplace=False)
47
+ )
48
+ (1): EncoderLayer(
49
+ (self_attn): MultiHeadedAttention(
50
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
51
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
52
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
53
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
54
+ (dropout): Dropout(p=0.2, inplace=False)
55
+ )
56
+ (feed_forward): MultiLayeredConv1d(
57
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
58
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
59
+ (dropout): Dropout(p=0.2, inplace=False)
60
+ )
61
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
62
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
63
+ (dropout): Dropout(p=0.2, inplace=False)
64
+ )
65
+ (2): EncoderLayer(
66
+ (self_attn): MultiHeadedAttention(
67
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
68
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
69
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
70
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
71
+ (dropout): Dropout(p=0.2, inplace=False)
72
+ )
73
+ (feed_forward): MultiLayeredConv1d(
74
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
75
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
76
+ (dropout): Dropout(p=0.2, inplace=False)
77
+ )
78
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
79
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
80
+ (dropout): Dropout(p=0.2, inplace=False)
81
+ )
82
+ (3): EncoderLayer(
83
+ (self_attn): MultiHeadedAttention(
84
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
85
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
86
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
87
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
88
+ (dropout): Dropout(p=0.2, inplace=False)
89
+ )
90
+ (feed_forward): MultiLayeredConv1d(
91
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
92
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
93
+ (dropout): Dropout(p=0.2, inplace=False)
94
+ )
95
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
96
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
97
+ (dropout): Dropout(p=0.2, inplace=False)
98
+ )
99
+ )
100
+ (after_norm): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
101
+ )
102
+ (duration_predictor): DurationPredictor(
103
+ (conv): ModuleList(
104
+ (0): Sequential(
105
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
106
+ (1): ReLU()
107
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
108
+ (3): Dropout(p=0.1, inplace=False)
109
+ )
110
+ (1): Sequential(
111
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
112
+ (1): ReLU()
113
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
114
+ (3): Dropout(p=0.1, inplace=False)
115
+ )
116
+ )
117
+ (linear): Linear(in_features=256, out_features=1, bias=True)
118
+ )
119
+ (pitch_predictor): VariancePredictor(
120
+ (conv): ModuleList(
121
+ (0): Sequential(
122
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
123
+ (1): ReLU()
124
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
125
+ (3): Dropout(p=0.5, inplace=False)
126
+ )
127
+ (1): Sequential(
128
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
129
+ (1): ReLU()
130
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
131
+ (3): Dropout(p=0.5, inplace=False)
132
+ )
133
+ (2): Sequential(
134
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
135
+ (1): ReLU()
136
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
137
+ (3): Dropout(p=0.5, inplace=False)
138
+ )
139
+ (3): Sequential(
140
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
141
+ (1): ReLU()
142
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
143
+ (3): Dropout(p=0.5, inplace=False)
144
+ )
145
+ (4): Sequential(
146
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
147
+ (1): ReLU()
148
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
149
+ (3): Dropout(p=0.5, inplace=False)
150
+ )
151
+ )
152
+ (linear): Linear(in_features=256, out_features=1, bias=True)
153
+ )
154
+ (pitch_embed): Sequential(
155
+ (0): Conv1d(1, 256, kernel_size=(1,), stride=(1,))
156
+ (1): Dropout(p=0.0, inplace=False)
157
+ )
158
+ (energy_predictor): VariancePredictor(
159
+ (conv): ModuleList(
160
+ (0): Sequential(
161
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
162
+ (1): ReLU()
163
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
164
+ (3): Dropout(p=0.5, inplace=False)
165
+ )
166
+ (1): Sequential(
167
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
168
+ (1): ReLU()
169
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
170
+ (3): Dropout(p=0.5, inplace=False)
171
+ )
172
+ )
173
+ (linear): Linear(in_features=256, out_features=1, bias=True)
174
+ )
175
+ (energy_embed): Sequential(
176
+ (0): Conv1d(1, 256, kernel_size=(1,), stride=(1,))
177
+ (1): Dropout(p=0.0, inplace=False)
178
+ )
179
+ (alignment_module): AlignmentModule(
180
+ (t_conv1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
181
+ (t_conv2): Conv1d(256, 256, kernel_size=(1,), stride=(1,))
182
+ (f_conv1): Conv1d(80, 256, kernel_size=(3,), stride=(1,), padding=(1,))
183
+ (f_conv2): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
184
+ (f_conv3): Conv1d(256, 256, kernel_size=(1,), stride=(1,))
185
+ )
186
+ (length_regulator): GaussianUpsampling()
187
+ (decoder): Encoder(
188
+ (embed): Sequential(
189
+ (0): ScaledPositionalEncoding(
190
+ (dropout): Dropout(p=0.2, inplace=False)
191
+ )
192
+ )
193
+ (encoders): MultiSequential(
194
+ (0): EncoderLayer(
195
+ (self_attn): MultiHeadedAttention(
196
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
197
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
198
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
199
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
200
+ (dropout): Dropout(p=0.2, inplace=False)
201
+ )
202
+ (feed_forward): MultiLayeredConv1d(
203
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
204
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
205
+ (dropout): Dropout(p=0.2, inplace=False)
206
+ )
207
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
208
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
209
+ (dropout): Dropout(p=0.2, inplace=False)
210
+ )
211
+ (1): EncoderLayer(
212
+ (self_attn): MultiHeadedAttention(
213
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
214
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
215
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
216
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
217
+ (dropout): Dropout(p=0.2, inplace=False)
218
+ )
219
+ (feed_forward): MultiLayeredConv1d(
220
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
221
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
222
+ (dropout): Dropout(p=0.2, inplace=False)
223
+ )
224
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
225
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
226
+ (dropout): Dropout(p=0.2, inplace=False)
227
+ )
228
+ (2): EncoderLayer(
229
+ (self_attn): MultiHeadedAttention(
230
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
231
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
232
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
233
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
234
+ (dropout): Dropout(p=0.2, inplace=False)
235
+ )
236
+ (feed_forward): MultiLayeredConv1d(
237
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
238
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
239
+ (dropout): Dropout(p=0.2, inplace=False)
240
+ )
241
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
242
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
243
+ (dropout): Dropout(p=0.2, inplace=False)
244
+ )
245
+ (3): EncoderLayer(
246
+ (self_attn): MultiHeadedAttention(
247
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
248
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
249
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
250
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
251
+ (dropout): Dropout(p=0.2, inplace=False)
252
+ )
253
+ (feed_forward): MultiLayeredConv1d(
254
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
255
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
256
+ (dropout): Dropout(p=0.2, inplace=False)
257
+ )
258
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
259
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
260
+ (dropout): Dropout(p=0.2, inplace=False)
261
+ )
262
+ )
263
+ (after_norm): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
264
+ )
265
+ (generator): HiFiGANGenerator(
266
+ (input_conv): Conv1d(256, 512, kernel_size=(7,), stride=(1,), padding=(3,))
267
+ (upsamples): ModuleList(
268
+ (0): Sequential(
269
+ (0): LeakyReLU(negative_slope=0.1)
270
+ (1): ConvTranspose1d(512, 256, kernel_size=(16,), stride=(8,), padding=(4,))
271
+ )
272
+ (1): Sequential(
273
+ (0): LeakyReLU(negative_slope=0.1)
274
+ (1): ConvTranspose1d(256, 128, kernel_size=(16,), stride=(8,), padding=(4,))
275
+ )
276
+ (2): Sequential(
277
+ (0): LeakyReLU(negative_slope=0.1)
278
+ (1): ConvTranspose1d(128, 64, kernel_size=(4,), stride=(2,), padding=(1,))
279
+ )
280
+ (3): Sequential(
281
+ (0): LeakyReLU(negative_slope=0.1)
282
+ (1): ConvTranspose1d(64, 32, kernel_size=(4,), stride=(2,), padding=(1,))
283
+ )
284
+ )
285
+ (blocks): ModuleList(
286
+ (0): ResidualBlock(
287
+ (convs1): ModuleList(
288
+ (0): Sequential(
289
+ (0): LeakyReLU(negative_slope=0.1)
290
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
291
+ )
292
+ (1): Sequential(
293
+ (0): LeakyReLU(negative_slope=0.1)
294
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
295
+ )
296
+ (2): Sequential(
297
+ (0): LeakyReLU(negative_slope=0.1)
298
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
299
+ )
300
+ )
301
+ (convs2): ModuleList(
302
+ (0): Sequential(
303
+ (0): LeakyReLU(negative_slope=0.1)
304
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
305
+ )
306
+ (1): Sequential(
307
+ (0): LeakyReLU(negative_slope=0.1)
308
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
309
+ )
310
+ (2): Sequential(
311
+ (0): LeakyReLU(negative_slope=0.1)
312
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
313
+ )
314
+ )
315
+ )
316
+ (1): ResidualBlock(
317
+ (convs1): ModuleList(
318
+ (0): Sequential(
319
+ (0): LeakyReLU(negative_slope=0.1)
320
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
321
+ )
322
+ (1): Sequential(
323
+ (0): LeakyReLU(negative_slope=0.1)
324
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
325
+ )
326
+ (2): Sequential(
327
+ (0): LeakyReLU(negative_slope=0.1)
328
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
329
+ )
330
+ )
331
+ (convs2): ModuleList(
332
+ (0): Sequential(
333
+ (0): LeakyReLU(negative_slope=0.1)
334
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
335
+ )
336
+ (1): Sequential(
337
+ (0): LeakyReLU(negative_slope=0.1)
338
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
339
+ )
340
+ (2): Sequential(
341
+ (0): LeakyReLU(negative_slope=0.1)
342
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
343
+ )
344
+ )
345
+ )
346
+ (2): ResidualBlock(
347
+ (convs1): ModuleList(
348
+ (0): Sequential(
349
+ (0): LeakyReLU(negative_slope=0.1)
350
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
351
+ )
352
+ (1): Sequential(
353
+ (0): LeakyReLU(negative_slope=0.1)
354
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
355
+ )
356
+ (2): Sequential(
357
+ (0): LeakyReLU(negative_slope=0.1)
358
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
359
+ )
360
+ )
361
+ (convs2): ModuleList(
362
+ (0): Sequential(
363
+ (0): LeakyReLU(negative_slope=0.1)
364
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
365
+ )
366
+ (1): Sequential(
367
+ (0): LeakyReLU(negative_slope=0.1)
368
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
369
+ )
370
+ (2): Sequential(
371
+ (0): LeakyReLU(negative_slope=0.1)
372
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
373
+ )
374
+ )
375
+ )
376
+ (3): ResidualBlock(
377
+ (convs1): ModuleList(
378
+ (0): Sequential(
379
+ (0): LeakyReLU(negative_slope=0.1)
380
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
381
+ )
382
+ (1): Sequential(
383
+ (0): LeakyReLU(negative_slope=0.1)
384
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
385
+ )
386
+ (2): Sequential(
387
+ (0): LeakyReLU(negative_slope=0.1)
388
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
389
+ )
390
+ )
391
+ (convs2): ModuleList(
392
+ (0): Sequential(
393
+ (0): LeakyReLU(negative_slope=0.1)
394
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
395
+ )
396
+ (1): Sequential(
397
+ (0): LeakyReLU(negative_slope=0.1)
398
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
399
+ )
400
+ (2): Sequential(
401
+ (0): LeakyReLU(negative_slope=0.1)
402
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
403
+ )
404
+ )
405
+ )
406
+ (4): ResidualBlock(
407
+ (convs1): ModuleList(
408
+ (0): Sequential(
409
+ (0): LeakyReLU(negative_slope=0.1)
410
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
411
+ )
412
+ (1): Sequential(
413
+ (0): LeakyReLU(negative_slope=0.1)
414
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
415
+ )
416
+ (2): Sequential(
417
+ (0): LeakyReLU(negative_slope=0.1)
418
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
419
+ )
420
+ )
421
+ (convs2): ModuleList(
422
+ (0): Sequential(
423
+ (0): LeakyReLU(negative_slope=0.1)
424
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
425
+ )
426
+ (1): Sequential(
427
+ (0): LeakyReLU(negative_slope=0.1)
428
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
429
+ )
430
+ (2): Sequential(
431
+ (0): LeakyReLU(negative_slope=0.1)
432
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
433
+ )
434
+ )
435
+ )
436
+ (5): ResidualBlock(
437
+ (convs1): ModuleList(
438
+ (0): Sequential(
439
+ (0): LeakyReLU(negative_slope=0.1)
440
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
441
+ )
442
+ (1): Sequential(
443
+ (0): LeakyReLU(negative_slope=0.1)
444
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
445
+ )
446
+ (2): Sequential(
447
+ (0): LeakyReLU(negative_slope=0.1)
448
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
449
+ )
450
+ )
451
+ (convs2): ModuleList(
452
+ (0): Sequential(
453
+ (0): LeakyReLU(negative_slope=0.1)
454
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
455
+ )
456
+ (1): Sequential(
457
+ (0): LeakyReLU(negative_slope=0.1)
458
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
459
+ )
460
+ (2): Sequential(
461
+ (0): LeakyReLU(negative_slope=0.1)
462
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
463
+ )
464
+ )
465
+ )
466
+ (6): ResidualBlock(
467
+ (convs1): ModuleList(
468
+ (0): Sequential(
469
+ (0): LeakyReLU(negative_slope=0.1)
470
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
471
+ )
472
+ (1): Sequential(
473
+ (0): LeakyReLU(negative_slope=0.1)
474
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
475
+ )
476
+ (2): Sequential(
477
+ (0): LeakyReLU(negative_slope=0.1)
478
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
479
+ )
480
+ )
481
+ (convs2): ModuleList(
482
+ (0): Sequential(
483
+ (0): LeakyReLU(negative_slope=0.1)
484
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
485
+ )
486
+ (1): Sequential(
487
+ (0): LeakyReLU(negative_slope=0.1)
488
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
489
+ )
490
+ (2): Sequential(
491
+ (0): LeakyReLU(negative_slope=0.1)
492
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
493
+ )
494
+ )
495
+ )
496
+ (7): ResidualBlock(
497
+ (convs1): ModuleList(
498
+ (0): Sequential(
499
+ (0): LeakyReLU(negative_slope=0.1)
500
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
501
+ )
502
+ (1): Sequential(
503
+ (0): LeakyReLU(negative_slope=0.1)
504
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
505
+ )
506
+ (2): Sequential(
507
+ (0): LeakyReLU(negative_slope=0.1)
508
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
509
+ )
510
+ )
511
+ (convs2): ModuleList(
512
+ (0): Sequential(
513
+ (0): LeakyReLU(negative_slope=0.1)
514
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
515
+ )
516
+ (1): Sequential(
517
+ (0): LeakyReLU(negative_slope=0.1)
518
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
519
+ )
520
+ (2): Sequential(
521
+ (0): LeakyReLU(negative_slope=0.1)
522
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
523
+ )
524
+ )
525
+ )
526
+ (8): ResidualBlock(
527
+ (convs1): ModuleList(
528
+ (0): Sequential(
529
+ (0): LeakyReLU(negative_slope=0.1)
530
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
531
+ )
532
+ (1): Sequential(
533
+ (0): LeakyReLU(negative_slope=0.1)
534
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
535
+ )
536
+ (2): Sequential(
537
+ (0): LeakyReLU(negative_slope=0.1)
538
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
539
+ )
540
+ )
541
+ (convs2): ModuleList(
542
+ (0): Sequential(
543
+ (0): LeakyReLU(negative_slope=0.1)
544
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
545
+ )
546
+ (1): Sequential(
547
+ (0): LeakyReLU(negative_slope=0.1)
548
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
549
+ )
550
+ (2): Sequential(
551
+ (0): LeakyReLU(negative_slope=0.1)
552
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
553
+ )
554
+ )
555
+ )
556
+ (9): ResidualBlock(
557
+ (convs1): ModuleList(
558
+ (0): Sequential(
559
+ (0): LeakyReLU(negative_slope=0.1)
560
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
561
+ )
562
+ (1): Sequential(
563
+ (0): LeakyReLU(negative_slope=0.1)
564
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
565
+ )
566
+ (2): Sequential(
567
+ (0): LeakyReLU(negative_slope=0.1)
568
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
569
+ )
570
+ )
571
+ (convs2): ModuleList(
572
+ (0): Sequential(
573
+ (0): LeakyReLU(negative_slope=0.1)
574
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
575
+ )
576
+ (1): Sequential(
577
+ (0): LeakyReLU(negative_slope=0.1)
578
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
579
+ )
580
+ (2): Sequential(
581
+ (0): LeakyReLU(negative_slope=0.1)
582
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
583
+ )
584
+ )
585
+ )
586
+ (10): ResidualBlock(
587
+ (convs1): ModuleList(
588
+ (0): Sequential(
589
+ (0): LeakyReLU(negative_slope=0.1)
590
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
591
+ )
592
+ (1): Sequential(
593
+ (0): LeakyReLU(negative_slope=0.1)
594
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
595
+ )
596
+ (2): Sequential(
597
+ (0): LeakyReLU(negative_slope=0.1)
598
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
599
+ )
600
+ )
601
+ (convs2): ModuleList(
602
+ (0): Sequential(
603
+ (0): LeakyReLU(negative_slope=0.1)
604
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
605
+ )
606
+ (1): Sequential(
607
+ (0): LeakyReLU(negative_slope=0.1)
608
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
609
+ )
610
+ (2): Sequential(
611
+ (0): LeakyReLU(negative_slope=0.1)
612
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
613
+ )
614
+ )
615
+ )
616
+ (11): ResidualBlock(
617
+ (convs1): ModuleList(
618
+ (0): Sequential(
619
+ (0): LeakyReLU(negative_slope=0.1)
620
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
621
+ )
622
+ (1): Sequential(
623
+ (0): LeakyReLU(negative_slope=0.1)
624
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
625
+ )
626
+ (2): Sequential(
627
+ (0): LeakyReLU(negative_slope=0.1)
628
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
629
+ )
630
+ )
631
+ (convs2): ModuleList(
632
+ (0): Sequential(
633
+ (0): LeakyReLU(negative_slope=0.1)
634
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
635
+ )
636
+ (1): Sequential(
637
+ (0): LeakyReLU(negative_slope=0.1)
638
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
639
+ )
640
+ (2): Sequential(
641
+ (0): LeakyReLU(negative_slope=0.1)
642
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
643
+ )
644
+ )
645
+ )
646
+ )
647
+ (output_conv): Sequential(
648
+ (0): LeakyReLU(negative_slope=0.01)
649
+ (1): Conv1d(32, 1, kernel_size=(7,), stride=(1,), padding=(3,))
650
+ (2): Tanh()
651
+ )
652
+ )
653
+ )
654
+ (discriminator): HiFiGANMultiScaleMultiPeriodDiscriminator(
655
+ (msd): HiFiGANMultiScaleDiscriminator(
656
+ (discriminators): ModuleList(
657
+ (0): HiFiGANScaleDiscriminator(
658
+ (layers): ModuleList(
659
+ (0): Sequential(
660
+ (0): Conv1d(1, 128, kernel_size=(15,), stride=(1,), padding=(7,))
661
+ (1): LeakyReLU(negative_slope=0.1)
662
+ )
663
+ (1): Sequential(
664
+ (0): Conv1d(128, 128, kernel_size=(41,), stride=(2,), padding=(20,), groups=4)
665
+ (1): LeakyReLU(negative_slope=0.1)
666
+ )
667
+ (2): Sequential(
668
+ (0): Conv1d(128, 256, kernel_size=(41,), stride=(2,), padding=(20,), groups=16)
669
+ (1): LeakyReLU(negative_slope=0.1)
670
+ )
671
+ (3): Sequential(
672
+ (0): Conv1d(256, 512, kernel_size=(41,), stride=(4,), padding=(20,), groups=16)
673
+ (1): LeakyReLU(negative_slope=0.1)
674
+ )
675
+ (4): Sequential(
676
+ (0): Conv1d(512, 1024, kernel_size=(41,), stride=(4,), padding=(20,), groups=16)
677
+ (1): LeakyReLU(negative_slope=0.1)
678
+ )
679
+ (5): Sequential(
680
+ (0): Conv1d(1024, 1024, kernel_size=(41,), stride=(1,), padding=(20,), groups=16)
681
+ (1): LeakyReLU(negative_slope=0.1)
682
+ )
683
+ (6): Sequential(
684
+ (0): Conv1d(1024, 1024, kernel_size=(5,), stride=(1,), padding=(2,))
685
+ (1): LeakyReLU(negative_slope=0.1)
686
+ )
687
+ (7): Conv1d(1024, 1, kernel_size=(3,), stride=(1,), padding=(1,))
688
+ )
689
+ )
690
+ )
691
+ )
692
+ (mpd): HiFiGANMultiPeriodDiscriminator(
693
+ (discriminators): ModuleList(
694
+ (0): HiFiGANPeriodDiscriminator(
695
+ (convs): ModuleList(
696
+ (0): Sequential(
697
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
698
+ (1): LeakyReLU(negative_slope=0.1)
699
+ )
700
+ (1): Sequential(
701
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
702
+ (1): LeakyReLU(negative_slope=0.1)
703
+ )
704
+ (2): Sequential(
705
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
706
+ (1): LeakyReLU(negative_slope=0.1)
707
+ )
708
+ (3): Sequential(
709
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
710
+ (1): LeakyReLU(negative_slope=0.1)
711
+ )
712
+ (4): Sequential(
713
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
714
+ (1): LeakyReLU(negative_slope=0.1)
715
+ )
716
+ )
717
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
718
+ )
719
+ (1): HiFiGANPeriodDiscriminator(
720
+ (convs): ModuleList(
721
+ (0): Sequential(
722
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
723
+ (1): LeakyReLU(negative_slope=0.1)
724
+ )
725
+ (1): Sequential(
726
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
727
+ (1): LeakyReLU(negative_slope=0.1)
728
+ )
729
+ (2): Sequential(
730
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
731
+ (1): LeakyReLU(negative_slope=0.1)
732
+ )
733
+ (3): Sequential(
734
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
735
+ (1): LeakyReLU(negative_slope=0.1)
736
+ )
737
+ (4): Sequential(
738
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
739
+ (1): LeakyReLU(negative_slope=0.1)
740
+ )
741
+ )
742
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
743
+ )
744
+ (2): HiFiGANPeriodDiscriminator(
745
+ (convs): ModuleList(
746
+ (0): Sequential(
747
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
748
+ (1): LeakyReLU(negative_slope=0.1)
749
+ )
750
+ (1): Sequential(
751
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
752
+ (1): LeakyReLU(negative_slope=0.1)
753
+ )
754
+ (2): Sequential(
755
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
756
+ (1): LeakyReLU(negative_slope=0.1)
757
+ )
758
+ (3): Sequential(
759
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
760
+ (1): LeakyReLU(negative_slope=0.1)
761
+ )
762
+ (4): Sequential(
763
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
764
+ (1): LeakyReLU(negative_slope=0.1)
765
+ )
766
+ )
767
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
768
+ )
769
+ (3): HiFiGANPeriodDiscriminator(
770
+ (convs): ModuleList(
771
+ (0): Sequential(
772
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
773
+ (1): LeakyReLU(negative_slope=0.1)
774
+ )
775
+ (1): Sequential(
776
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
777
+ (1): LeakyReLU(negative_slope=0.1)
778
+ )
779
+ (2): Sequential(
780
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
781
+ (1): LeakyReLU(negative_slope=0.1)
782
+ )
783
+ (3): Sequential(
784
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
785
+ (1): LeakyReLU(negative_slope=0.1)
786
+ )
787
+ (4): Sequential(
788
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
789
+ (1): LeakyReLU(negative_slope=0.1)
790
+ )
791
+ )
792
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
793
+ )
794
+ (4): HiFiGANPeriodDiscriminator(
795
+ (convs): ModuleList(
796
+ (0): Sequential(
797
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
798
+ (1): LeakyReLU(negative_slope=0.1)
799
+ )
800
+ (1): Sequential(
801
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
802
+ (1): LeakyReLU(negative_slope=0.1)
803
+ )
804
+ (2): Sequential(
805
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
806
+ (1): LeakyReLU(negative_slope=0.1)
807
+ )
808
+ (3): Sequential(
809
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
810
+ (1): LeakyReLU(negative_slope=0.1)
811
+ )
812
+ (4): Sequential(
813
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
814
+ (1): LeakyReLU(negative_slope=0.1)
815
+ )
816
+ )
817
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
818
+ )
819
+ )
820
+ )
821
+ )
822
+ (generator_adv_loss): GeneratorAdversarialLoss()
823
+ (discriminator_adv_loss): DiscriminatorAdversarialLoss()
824
+ (feat_match_loss): FeatureMatchLoss()
825
+ (mel_loss): MelSpectrogramLoss(
826
+ (wav_to_mel): LogMelFbank(
827
+ (stft): Stft(n_fft=1024, win_length=1024, hop_length=256, center=True, normalized=False, onesided=True)
828
+ (logmel): LogMel(sr=24000, n_fft=1024, n_mels=80, fmin=0, fmax=12000.0, htk=False)
829
+ )
830
+ )
831
+ (var_loss): VarianceLoss(
832
+ (mse_criterion): MSELoss()
833
+ (duration_criterion): DurationPredictorLoss(
834
+ (criterion): MSELoss()
835
+ )
836
+ )
837
+ (forwardsum_loss): ForwardSumLoss()
838
+ )
839
+ )
840
+
841
+ Model summary:
842
+ Class Name: ESPnetGANTTSModel
843
+ Total Number of model parameters: 83.28 M
844
+ Number of trainable parameters: 83.28 M (100.0%)
845
+ Size: 333.11 MB
846
+ Type: torch.float32
847
+ [92b100c97f43] 2025-03-04 21:23:31,686 (abs_task:1161) INFO: Optimizer:
848
+ AdamW (
849
+ Parameter Group 0
850
+ amsgrad: False
851
+ betas: [0.8, 0.99]
852
+ eps: 1e-09
853
+ initial_lr: 0.0002
854
+ lr: 0.0002
855
+ weight_decay: 0.0
856
+ )
857
+ [92b100c97f43] 2025-03-04 21:23:31,686 (abs_task:1162) INFO: Scheduler: <torch.optim.lr_scheduler.ExponentialLR object at 0x7fea878fb160>
858
+ [92b100c97f43] 2025-03-04 21:23:31,686 (abs_task:1161) INFO: Optimizer2:
859
+ AdamW (
860
+ Parameter Group 0
861
+ amsgrad: False
862
+ betas: [0.8, 0.99]
863
+ eps: 1e-09
864
+ initial_lr: 0.0002
865
+ lr: 0.0002
866
+ weight_decay: 0.0
867
+ )
868
+ [92b100c97f43] 2025-03-04 21:23:31,686 (abs_task:1162) INFO: Scheduler2: <torch.optim.lr_scheduler.ExponentialLR object at 0x7fea9681f280>
869
+ [92b100c97f43] 2025-03-04 21:23:31,686 (abs_task:1171) INFO: Saving the configuration in exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/config.yaml
870
+ [92b100c97f43] 2025-03-04 21:23:31,705 (abs_task:1182) INFO: Namespace(accum_grad=1, allow_variable_data_keys=False, batch_bins=6000000, batch_size=20, batch_type='numel', best_model_criterion=[['valid', 'text2mel_loss', 'min'], ['train', 'text2mel_loss', 'min'], ['train', 'total_count', 'max']], bpemodel=None, chunk_length=500, chunk_shift_ratio=0.5, cleaner='jaconv', collect_stats=True, config='conf/tuning/train_jets.yaml', cudnn_benchmark=False, cudnn_deterministic=False, cudnn_enabled=True, detect_anomaly=False, dist_backend='nccl', dist_init_method='env://', dist_launcher=None, dist_master_addr=None, dist_master_port=None, dist_rank=None, dist_world_size=None, distributed=False, dry_run=False, early_stopping_criterion=('valid', 'loss', 'min'), energy_extract='energy', energy_extract_conf={'reduction_factor': 1, 'use_token_averaged_energy': False, 'fs': 24000, 'n_fft': 2048, 'hop_length': 300, 'win_length': 1200}, energy_normalize=None, energy_normalize_conf={}, feats_extract='fbank', feats_extract_conf={'n_fft': 2048, 'hop_length': 300, 'win_length': 1200, 'fs': 24000, 'fmin': 80, 'fmax': 7600, 'n_mels': 80}, fold_length=[], freeze_param=[], g2p='pyopenjtalk', generator_first=True, grad_clip=-1, grad_clip_type=2.0, grad_noise=False, ignore_init_mismatch=False, init_param=[], iterator_type='sequence', keep_nbest_models=-1, local_rank=None, log_interval=50, log_level='INFO', max_cache_fd=32, max_cache_size=0.0, max_epoch=130, model_conf={}, multiple_iterator=False, multiprocessing_distributed=False, nbest_averaging_interval=0, ngpu=0, no_forward_run=False, non_linguistic_symbols=None, normalize=None, normalize_conf={}, num_att_plot=3, num_cache_chunks=1024, num_iters_per_epoch=1000, num_workers=16, odim=None, optim='adamw', optim2='adamw', optim2_conf={'lr': 0.0002, 'betas': [0.8, 0.99], 'eps': 1e-09, 'weight_decay': 0.0}, optim_conf={'lr': 0.0002, 'betas': [0.8, 0.99], 'eps': 1e-09, 'weight_decay': 0.0}, output_dir='exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1', patience=None, pitch_extract='dio', pitch_extract_conf={'reduction_factor': 1, 'use_token_averaged_f0': False, 'fs': 24000, 'n_fft': 2048, 'hop_length': 300, 'f0max': 400, 'f0min': 80}, pitch_normalize=None, pitch_normalize_conf={}, pretrain_path=None, print_config=False, required=['output_dir', 'token_list'], resume=False, scheduler='exponentiallr', scheduler2='exponentiallr', scheduler2_conf={'gamma': 0.999875}, scheduler_conf={'gamma': 0.999875}, seed=777, sharded_ddp=False, sort_batch='descending', sort_in_batch='descending', token_list=['<blank>', '<unk>', 'o', 'a', 'u', 'i', 'e', 'k', 'r', 't', 'n', 'pau', 'N', 's', 'sh', 'd', 'm', 'g', 'w', 'b', 'cl', 'I', 'j', 'ch', 'y', 'U', 'h', 'p', 'ts', 'f', 'z', 'ky', 'ny', 'gy', 'ry', 'hy', 'my', 'by', 'py', 'v', '<sos/eos>'], token_type='phn', train_data_path_and_name_and_type=[('dump/raw/jvs010_tr_no_dev/text', 'text', 'text'), ('dump/raw/jvs010_tr_no_dev/wav.scp', 'speech', 'sound')], train_dtype='float32', train_shape_file=['exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.1.scp'], tts='jets', tts_conf={'generator_type': 'jets_generator', 'generator_params': {'adim': 256, 'aheads': 2, 'elayers': 4, 'eunits': 1024, 'dlayers': 4, 'dunits': 1024, 'positionwise_layer_type': 'conv1d', 'positionwise_conv_kernel_size': 3, 'duration_predictor_layers': 2, 'duration_predictor_chans': 256, 'duration_predictor_kernel_size': 3, 'use_masking': True, 'encoder_normalize_before': True, 'decoder_normalize_before': True, 'encoder_type': 'transformer', 'decoder_type': 'transformer', 'conformer_rel_pos_type': 'latest', 'conformer_pos_enc_layer_type': 'rel_pos', 'conformer_self_attn_layer_type': 'rel_selfattn', 'conformer_activation_type': 'swish', 'use_macaron_style_in_conformer': True, 'use_cnn_in_conformer': True, 'conformer_enc_kernel_size': 7, 'conformer_dec_kernel_size': 31, 'init_type': 'xavier_uniform', 'transformer_enc_dropout_rate': 0.2, 'transformer_enc_positional_dropout_rate': 0.2, 'transformer_enc_attn_dropout_rate': 0.2, 'transformer_dec_dropout_rate': 0.2, 'transformer_dec_positional_dropout_rate': 0.2, 'transformer_dec_attn_dropout_rate': 0.2, 'pitch_predictor_layers': 5, 'pitch_predictor_chans': 256, 'pitch_predictor_kernel_size': 5, 'pitch_predictor_dropout': 0.5, 'pitch_embed_kernel_size': 1, 'pitch_embed_dropout': 0.0, 'stop_gradient_from_pitch_predictor': True, 'energy_predictor_layers': 2, 'energy_predictor_chans': 256, 'energy_predictor_kernel_size': 3, 'energy_predictor_dropout': 0.5, 'energy_embed_kernel_size': 1, 'energy_embed_dropout': 0.0, 'stop_gradient_from_energy_predictor': False, 'generator_out_channels': 1, 'generator_channels': 512, 'generator_global_channels': -1, 'generator_kernel_size': 7, 'generator_upsample_scales': [8, 8, 2, 2], 'generator_upsample_kernel_sizes': [16, 16, 4, 4], 'generator_resblock_kernel_sizes': [3, 7, 11], 'generator_resblock_dilations': [[1, 3, 5], [1, 3, 5], [1, 3, 5]], 'generator_use_additional_convs': True, 'generator_bias': True, 'generator_nonlinear_activation': 'LeakyReLU', 'generator_nonlinear_activation_params': {'negative_slope': 0.1}, 'generator_use_weight_norm': True, 'segment_size': 64, 'idim': 41, 'odim': 80}, 'discriminator_type': 'hifigan_multi_scale_multi_period_discriminator', 'discriminator_params': {'scales': 1, 'scale_downsample_pooling': 'AvgPool1d', 'scale_downsample_pooling_params': {'kernel_size': 4, 'stride': 2, 'padding': 2}, 'scale_discriminator_params': {'in_channels': 1, 'out_channels': 1, 'kernel_sizes': [15, 41, 5, 3], 'channels': 128, 'max_downsample_channels': 1024, 'max_groups': 16, 'bias': True, 'downsample_scales': [2, 2, 4, 4, 1], 'nonlinear_activation': 'LeakyReLU', 'nonlinear_activation_params': {'negative_slope': 0.1}, 'use_weight_norm': True, 'use_spectral_norm': False}, 'follow_official_norm': False, 'periods': [2, 3, 5, 7, 11], 'period_discriminator_params': {'in_channels': 1, 'out_channels': 1, 'kernel_sizes': [5, 3], 'channels': 32, 'downsample_scales': [3, 3, 3, 3, 1], 'max_downsample_channels': 1024, 'bias': True, 'nonlinear_activation': 'LeakyReLU', 'nonlinear_activation_params': {'negative_slope': 0.1}, 'use_weight_norm': True, 'use_spectral_norm': False}}, 'generator_adv_loss_params': {'average_by_discriminators': False, 'loss_type': 'mse'}, 'discriminator_adv_loss_params': {'average_by_discriminators': False, 'loss_type': 'mse'}, 'feat_match_loss_params': {'average_by_discriminators': False, 'average_by_layers': False, 'include_final_outputs': True}, 'mel_loss_params': {'fs': 24000, 'n_fft': 1024, 'hop_length': 256, 'win_length': None, 'window': 'hann', 'n_mels': 80, 'fmin': 0, 'fmax': None, 'log_base': None}, 'lambda_adv': 1.0, 'lambda_mel': 45.0, 'lambda_feat_match': 2.0, 'lambda_var': 1.0, 'lambda_align': 2.0, 'sampling_rate': 24000, 'cache_generator_outputs': True}, unused_parameters=True, use_amp=False, use_matplotlib=True, use_preprocessor=True, use_tensorboard=True, use_wandb=False, val_scheduler_criterion=('valid', 'loss'), valid_batch_bins=None, valid_batch_size=None, valid_batch_type=None, valid_data_path_and_name_and_type=[('dump/raw/jvs010_dev/text', 'text', 'text'), ('dump/raw/jvs010_dev/wav.scp', 'speech', 'sound')], valid_max_cache_size=None, valid_shape_file=['exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.1.scp'], version='202204', wandb_entity=None, wandb_id=None, wandb_model_log_interval=-1, wandb_name=None, wandb_project=None, write_collected_feats=False)
871
+ /work/espnet/espnet2/layers/stft.py:166: UserWarning: __floordiv__ is deprecated, and its behavior will change in a future version of pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). This results in incorrect rounding for negative values. To keep the current behavior, use torch.div(a, b, rounding_mode='trunc'), or for actual floor division, use torch.div(a, b, rounding_mode='floor').
872
+ olens = (ilens - self.n_fft) // self.hop_length + 1
873
+ # Accounting: time=11 threads=1
874
+ # Ended (code 0) at Tue Mar 4 21:23:37 JST 2025, elapsed time 11 seconds
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/config.yaml ADDED
@@ -0,0 +1,386 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ config: conf/tuning/train_jets.yaml
2
+ print_config: false
3
+ log_level: INFO
4
+ dry_run: false
5
+ iterator_type: sequence
6
+ output_dir: exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1
7
+ ngpu: 0
8
+ seed: 777
9
+ num_workers: 16
10
+ num_att_plot: 3
11
+ dist_backend: nccl
12
+ dist_init_method: env://
13
+ dist_world_size: null
14
+ dist_rank: null
15
+ local_rank: null
16
+ dist_master_addr: null
17
+ dist_master_port: null
18
+ dist_launcher: null
19
+ multiprocessing_distributed: false
20
+ unused_parameters: true
21
+ sharded_ddp: false
22
+ cudnn_enabled: true
23
+ cudnn_benchmark: false
24
+ cudnn_deterministic: false
25
+ collect_stats: true
26
+ write_collected_feats: false
27
+ max_epoch: 130
28
+ patience: null
29
+ val_scheduler_criterion:
30
+ - valid
31
+ - loss
32
+ early_stopping_criterion:
33
+ - valid
34
+ - loss
35
+ - min
36
+ best_model_criterion:
37
+ - - valid
38
+ - text2mel_loss
39
+ - min
40
+ - - train
41
+ - text2mel_loss
42
+ - min
43
+ - - train
44
+ - total_count
45
+ - max
46
+ keep_nbest_models: -1
47
+ nbest_averaging_interval: 0
48
+ grad_clip: -1
49
+ grad_clip_type: 2.0
50
+ grad_noise: false
51
+ accum_grad: 1
52
+ no_forward_run: false
53
+ resume: false
54
+ train_dtype: float32
55
+ use_amp: false
56
+ log_interval: 50
57
+ use_matplotlib: true
58
+ use_tensorboard: true
59
+ use_wandb: false
60
+ wandb_project: null
61
+ wandb_id: null
62
+ wandb_entity: null
63
+ wandb_name: null
64
+ wandb_model_log_interval: -1
65
+ detect_anomaly: false
66
+ pretrain_path: null
67
+ init_param: []
68
+ ignore_init_mismatch: false
69
+ freeze_param: []
70
+ num_iters_per_epoch: 1000
71
+ batch_size: 20
72
+ valid_batch_size: null
73
+ batch_bins: 6000000
74
+ valid_batch_bins: null
75
+ train_shape_file:
76
+ - exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.1.scp
77
+ valid_shape_file:
78
+ - exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.1.scp
79
+ batch_type: numel
80
+ valid_batch_type: null
81
+ fold_length: []
82
+ sort_in_batch: descending
83
+ sort_batch: descending
84
+ multiple_iterator: false
85
+ chunk_length: 500
86
+ chunk_shift_ratio: 0.5
87
+ num_cache_chunks: 1024
88
+ train_data_path_and_name_and_type:
89
+ - - dump/raw/jvs010_tr_no_dev/text
90
+ - text
91
+ - text
92
+ - - dump/raw/jvs010_tr_no_dev/wav.scp
93
+ - speech
94
+ - sound
95
+ valid_data_path_and_name_and_type:
96
+ - - dump/raw/jvs010_dev/text
97
+ - text
98
+ - text
99
+ - - dump/raw/jvs010_dev/wav.scp
100
+ - speech
101
+ - sound
102
+ allow_variable_data_keys: false
103
+ max_cache_size: 0.0
104
+ max_cache_fd: 32
105
+ valid_max_cache_size: null
106
+ optim: adamw
107
+ optim_conf:
108
+ lr: 0.0002
109
+ betas:
110
+ - 0.8
111
+ - 0.99
112
+ eps: 1.0e-09
113
+ weight_decay: 0.0
114
+ scheduler: exponentiallr
115
+ scheduler_conf:
116
+ gamma: 0.999875
117
+ optim2: adamw
118
+ optim2_conf:
119
+ lr: 0.0002
120
+ betas:
121
+ - 0.8
122
+ - 0.99
123
+ eps: 1.0e-09
124
+ weight_decay: 0.0
125
+ scheduler2: exponentiallr
126
+ scheduler2_conf:
127
+ gamma: 0.999875
128
+ generator_first: true
129
+ token_list:
130
+ - <blank>
131
+ - <unk>
132
+ - o
133
+ - a
134
+ - u
135
+ - i
136
+ - e
137
+ - k
138
+ - r
139
+ - t
140
+ - n
141
+ - pau
142
+ - N
143
+ - s
144
+ - sh
145
+ - d
146
+ - m
147
+ - g
148
+ - w
149
+ - b
150
+ - cl
151
+ - I
152
+ - j
153
+ - ch
154
+ - y
155
+ - U
156
+ - h
157
+ - p
158
+ - ts
159
+ - f
160
+ - z
161
+ - ky
162
+ - ny
163
+ - gy
164
+ - ry
165
+ - hy
166
+ - my
167
+ - by
168
+ - py
169
+ - v
170
+ - <sos/eos>
171
+ odim: null
172
+ model_conf: {}
173
+ use_preprocessor: true
174
+ token_type: phn
175
+ bpemodel: null
176
+ non_linguistic_symbols: null
177
+ cleaner: jaconv
178
+ g2p: pyopenjtalk
179
+ feats_extract: fbank
180
+ feats_extract_conf:
181
+ n_fft: 2048
182
+ hop_length: 300
183
+ win_length: 1200
184
+ fs: 24000
185
+ fmin: 80
186
+ fmax: 7600
187
+ n_mels: 80
188
+ normalize: null
189
+ normalize_conf: {}
190
+ tts: jets
191
+ tts_conf:
192
+ generator_type: jets_generator
193
+ generator_params:
194
+ adim: 256
195
+ aheads: 2
196
+ elayers: 4
197
+ eunits: 1024
198
+ dlayers: 4
199
+ dunits: 1024
200
+ positionwise_layer_type: conv1d
201
+ positionwise_conv_kernel_size: 3
202
+ duration_predictor_layers: 2
203
+ duration_predictor_chans: 256
204
+ duration_predictor_kernel_size: 3
205
+ use_masking: true
206
+ encoder_normalize_before: true
207
+ decoder_normalize_before: true
208
+ encoder_type: transformer
209
+ decoder_type: transformer
210
+ conformer_rel_pos_type: latest
211
+ conformer_pos_enc_layer_type: rel_pos
212
+ conformer_self_attn_layer_type: rel_selfattn
213
+ conformer_activation_type: swish
214
+ use_macaron_style_in_conformer: true
215
+ use_cnn_in_conformer: true
216
+ conformer_enc_kernel_size: 7
217
+ conformer_dec_kernel_size: 31
218
+ init_type: xavier_uniform
219
+ transformer_enc_dropout_rate: 0.2
220
+ transformer_enc_positional_dropout_rate: 0.2
221
+ transformer_enc_attn_dropout_rate: 0.2
222
+ transformer_dec_dropout_rate: 0.2
223
+ transformer_dec_positional_dropout_rate: 0.2
224
+ transformer_dec_attn_dropout_rate: 0.2
225
+ pitch_predictor_layers: 5
226
+ pitch_predictor_chans: 256
227
+ pitch_predictor_kernel_size: 5
228
+ pitch_predictor_dropout: 0.5
229
+ pitch_embed_kernel_size: 1
230
+ pitch_embed_dropout: 0.0
231
+ stop_gradient_from_pitch_predictor: true
232
+ energy_predictor_layers: 2
233
+ energy_predictor_chans: 256
234
+ energy_predictor_kernel_size: 3
235
+ energy_predictor_dropout: 0.5
236
+ energy_embed_kernel_size: 1
237
+ energy_embed_dropout: 0.0
238
+ stop_gradient_from_energy_predictor: false
239
+ generator_out_channels: 1
240
+ generator_channels: 512
241
+ generator_global_channels: -1
242
+ generator_kernel_size: 7
243
+ generator_upsample_scales:
244
+ - 8
245
+ - 8
246
+ - 2
247
+ - 2
248
+ generator_upsample_kernel_sizes:
249
+ - 16
250
+ - 16
251
+ - 4
252
+ - 4
253
+ generator_resblock_kernel_sizes:
254
+ - 3
255
+ - 7
256
+ - 11
257
+ generator_resblock_dilations:
258
+ - - 1
259
+ - 3
260
+ - 5
261
+ - - 1
262
+ - 3
263
+ - 5
264
+ - - 1
265
+ - 3
266
+ - 5
267
+ generator_use_additional_convs: true
268
+ generator_bias: true
269
+ generator_nonlinear_activation: LeakyReLU
270
+ generator_nonlinear_activation_params:
271
+ negative_slope: 0.1
272
+ generator_use_weight_norm: true
273
+ segment_size: 64
274
+ idim: 41
275
+ odim: 80
276
+ discriminator_type: hifigan_multi_scale_multi_period_discriminator
277
+ discriminator_params:
278
+ scales: 1
279
+ scale_downsample_pooling: AvgPool1d
280
+ scale_downsample_pooling_params:
281
+ kernel_size: 4
282
+ stride: 2
283
+ padding: 2
284
+ scale_discriminator_params:
285
+ in_channels: 1
286
+ out_channels: 1
287
+ kernel_sizes:
288
+ - 15
289
+ - 41
290
+ - 5
291
+ - 3
292
+ channels: 128
293
+ max_downsample_channels: 1024
294
+ max_groups: 16
295
+ bias: true
296
+ downsample_scales:
297
+ - 2
298
+ - 2
299
+ - 4
300
+ - 4
301
+ - 1
302
+ nonlinear_activation: LeakyReLU
303
+ nonlinear_activation_params:
304
+ negative_slope: 0.1
305
+ use_weight_norm: true
306
+ use_spectral_norm: false
307
+ follow_official_norm: false
308
+ periods:
309
+ - 2
310
+ - 3
311
+ - 5
312
+ - 7
313
+ - 11
314
+ period_discriminator_params:
315
+ in_channels: 1
316
+ out_channels: 1
317
+ kernel_sizes:
318
+ - 5
319
+ - 3
320
+ channels: 32
321
+ downsample_scales:
322
+ - 3
323
+ - 3
324
+ - 3
325
+ - 3
326
+ - 1
327
+ max_downsample_channels: 1024
328
+ bias: true
329
+ nonlinear_activation: LeakyReLU
330
+ nonlinear_activation_params:
331
+ negative_slope: 0.1
332
+ use_weight_norm: true
333
+ use_spectral_norm: false
334
+ generator_adv_loss_params:
335
+ average_by_discriminators: false
336
+ loss_type: mse
337
+ discriminator_adv_loss_params:
338
+ average_by_discriminators: false
339
+ loss_type: mse
340
+ feat_match_loss_params:
341
+ average_by_discriminators: false
342
+ average_by_layers: false
343
+ include_final_outputs: true
344
+ mel_loss_params:
345
+ fs: 24000
346
+ n_fft: 1024
347
+ hop_length: 256
348
+ win_length: null
349
+ window: hann
350
+ n_mels: 80
351
+ fmin: 0
352
+ fmax: null
353
+ log_base: null
354
+ lambda_adv: 1.0
355
+ lambda_mel: 45.0
356
+ lambda_feat_match: 2.0
357
+ lambda_var: 1.0
358
+ lambda_align: 2.0
359
+ sampling_rate: 24000
360
+ cache_generator_outputs: true
361
+ pitch_extract: dio
362
+ pitch_extract_conf:
363
+ reduction_factor: 1
364
+ use_token_averaged_f0: false
365
+ fs: 24000
366
+ n_fft: 2048
367
+ hop_length: 300
368
+ f0max: 400
369
+ f0min: 80
370
+ pitch_normalize: null
371
+ pitch_normalize_conf: {}
372
+ energy_extract: energy
373
+ energy_extract_conf:
374
+ reduction_factor: 1
375
+ use_token_averaged_energy: false
376
+ fs: 24000
377
+ n_fft: 2048
378
+ hop_length: 300
379
+ win_length: 1200
380
+ energy_normalize: null
381
+ energy_normalize_conf: {}
382
+ required:
383
+ - output_dir
384
+ - token_list
385
+ version: '202204'
386
+ distributed: false
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/batch_keys ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ text
2
+ speech
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/energy_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71835ae88e42963ae5e3de88a9c34a4e434c2dd2662455b4e61f347e2bd7f6f2
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/energy_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5541947d077b0041a3487937ee9145a25d632bc5bbd43711f977b48ac2569c6b
3
+ size 770
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/feats_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71835ae88e42963ae5e3de88a9c34a4e434c2dd2662455b4e61f347e2bd7f6f2
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/feats_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc5a955096b7b154b0e20365fd4a1cae7cc6fdf0e5cc0181abc8ca0dbba542af
3
+ size 1402
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/pitch_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71835ae88e42963ae5e3de88a9c34a4e434c2dd2662455b4e61f347e2bd7f6f2
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/pitch_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:802e7de46e59805b1ba5263f3208ccfc91ab4cd6c1ca757b7ec507f8f3d07bcc
3
+ size 770
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/speech_shape ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ jvs010_VOICEACTRESS100_001 178800
2
+ jvs010_VOICEACTRESS100_002 198240
3
+ jvs010_VOICEACTRESS100_003 128640
4
+ jvs010_VOICEACTRESS100_004 132000
5
+ jvs010_VOICEACTRESS100_005 277440
6
+ jvs010_VOICEACTRESS100_006 94560
7
+ jvs010_VOICEACTRESS100_007 182160
8
+ jvs010_VOICEACTRESS100_008 180960
9
+ jvs010_VOICEACTRESS100_009 145920
10
+ jvs010_VOICEACTRESS100_010 105359
11
+ jvs010_VOICEACTRESS100_011 148080
12
+ jvs010_VOICEACTRESS100_012 130320
13
+ jvs010_VOICEACTRESS100_013 117839
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/stats_keys ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ feats
2
+ feats_lengths
3
+ pitch
4
+ pitch_lengths
5
+ energy
6
+ energy_lengths
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/train/text_shape ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ jvs010_VOICEACTRESS100_001 78
2
+ jvs010_VOICEACTRESS100_002 91
3
+ jvs010_VOICEACTRESS100_003 69
4
+ jvs010_VOICEACTRESS100_004 64
5
+ jvs010_VOICEACTRESS100_005 121
6
+ jvs010_VOICEACTRESS100_006 49
7
+ jvs010_VOICEACTRESS100_007 93
8
+ jvs010_VOICEACTRESS100_008 82
9
+ jvs010_VOICEACTRESS100_009 69
10
+ jvs010_VOICEACTRESS100_010 49
11
+ jvs010_VOICEACTRESS100_011 77
12
+ jvs010_VOICEACTRESS100_012 55
13
+ jvs010_VOICEACTRESS100_013 54
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/batch_keys ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ text
2
+ speech
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/energy_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:149732b1e4eb869bc4896e15295f8daa910f81fa02a9f50c1821473ebedc9e3f
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/energy_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4931a92d9fcb8b8b598841c3589ac0e11b838be56304fd37301a3fc02a30bff5
3
+ size 770
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/feats_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:149732b1e4eb869bc4896e15295f8daa910f81fa02a9f50c1821473ebedc9e3f
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/feats_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc879120415bfea6b9b8b9559285f9b75cc8881682d326559b920626c778332e
3
+ size 1402
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/pitch_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:149732b1e4eb869bc4896e15295f8daa910f81fa02a9f50c1821473ebedc9e3f
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/pitch_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e277313a3d955098f9f250192f18bce2b31d2a442dc9e8a3ac909b9238e66c1
3
+ size 770
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/speech_shape ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ jvs010_BASIC5000_0113 143040
2
+ jvs010_BASIC5000_0261 77520
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/stats_keys ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ feats
2
+ feats_lengths
3
+ pitch
4
+ pitch_lengths
5
+ energy
6
+ energy_lengths
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.1/valid/text_shape ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ jvs010_BASIC5000_0113 56
2
+ jvs010_BASIC5000_0261 39
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2.log ADDED
@@ -0,0 +1,874 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # python3 -m espnet2.bin.gan_tts_train --collect_stats true --write_collected_feats false --use_preprocessor true --token_type phn --token_list dump/token_list/phn_jaconv_pyopenjtalk/tokens.txt --non_linguistic_symbols none --cleaner jaconv --g2p pyopenjtalk --normalize none --pitch_normalize none --energy_normalize none --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/text,text,text --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/wav.scp,speech,sound --valid_data_path_and_name_and_type dump/raw/jvs010_dev/text,text,text --valid_data_path_and_name_and_type dump/raw/jvs010_dev/wav.scp,speech,sound --train_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.2.scp --valid_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.2.scp --output_dir exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2 --config conf/tuning/train_jets.yaml --feats_extract fbank --feats_extract_conf n_fft=2048 --feats_extract_conf hop_length=300 --feats_extract_conf win_length=1200 --feats_extract_conf fs=24000 --feats_extract_conf fmin=80 --feats_extract_conf fmax=7600 --feats_extract_conf n_mels=80 --pitch_extract_conf fs=24000 --pitch_extract_conf n_fft=2048 --pitch_extract_conf hop_length=300 --pitch_extract_conf f0max=400 --pitch_extract_conf f0min=80 --energy_extract_conf fs=24000 --energy_extract_conf n_fft=2048 --energy_extract_conf hop_length=300 --energy_extract_conf win_length=1200
2
+ # Started at Tue Mar 4 21:23:26 JST 2025
3
+ #
4
+ /usr/lib/python3/dist-packages/requests/__init__.py:89: RequestsDependencyWarning: urllib3 (2.2.3) or chardet (3.0.4) doesn't match a supported version!
5
+ warnings.warn("urllib3 ({}) or chardet ({}) doesn't match a supported "
6
+ /usr/bin/python3 /work/espnet/espnet2/bin/gan_tts_train.py --collect_stats true --write_collected_feats false --use_preprocessor true --token_type phn --token_list dump/token_list/phn_jaconv_pyopenjtalk/tokens.txt --non_linguistic_symbols none --cleaner jaconv --g2p pyopenjtalk --normalize none --pitch_normalize none --energy_normalize none --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/text,text,text --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/wav.scp,speech,sound --valid_data_path_and_name_and_type dump/raw/jvs010_dev/text,text,text --valid_data_path_and_name_and_type dump/raw/jvs010_dev/wav.scp,speech,sound --train_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.2.scp --valid_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.2.scp --output_dir exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2 --config conf/tuning/train_jets.yaml --feats_extract fbank --feats_extract_conf n_fft=2048 --feats_extract_conf hop_length=300 --feats_extract_conf win_length=1200 --feats_extract_conf fs=24000 --feats_extract_conf fmin=80 --feats_extract_conf fmax=7600 --feats_extract_conf n_mels=80 --pitch_extract_conf fs=24000 --pitch_extract_conf n_fft=2048 --pitch_extract_conf hop_length=300 --pitch_extract_conf f0max=400 --pitch_extract_conf f0min=80 --energy_extract_conf fs=24000 --energy_extract_conf n_fft=2048 --energy_extract_conf hop_length=300 --energy_extract_conf win_length=1200
7
+ [92b100c97f43] 2025-03-04 21:23:29,163 (gan_tts:304) INFO: Vocabulary size: 41
8
+ [92b100c97f43] 2025-03-04 21:23:29,385 (encoder:172) INFO: encoder self-attention layer type = self-attention
9
+ [92b100c97f43] 2025-03-04 21:23:29,507 (encoder:172) INFO: encoder self-attention layer type = self-attention
10
+ [92b100c97f43] 2025-03-04 21:23:31,643 (abs_task:1157) INFO: pytorch.version=1.10.1+cu113, cuda.available=True, cudnn.version=8200, cudnn.benchmark=False, cudnn.deterministic=False
11
+ [92b100c97f43] 2025-03-04 21:23:31,653 (abs_task:1158) INFO: Model structure:
12
+ ESPnetGANTTSModel(
13
+ (feats_extract): LogMelFbank(
14
+ (stft): Stft(n_fft=2048, win_length=1200, hop_length=300, center=True, normalized=False, onesided=True)
15
+ (logmel): LogMel(sr=24000, n_fft=2048, n_mels=80, fmin=80, fmax=7600, htk=False)
16
+ )
17
+ (pitch_extract): Dio()
18
+ (energy_extract): Energy(
19
+ (stft): Stft(n_fft=2048, win_length=1200, hop_length=300, center=True, normalized=False, onesided=True)
20
+ )
21
+ (tts): JETS(
22
+ (generator): JETSGenerator(
23
+ (encoder): Encoder(
24
+ (embed): Sequential(
25
+ (0): Embedding(41, 256, padding_idx=0)
26
+ (1): ScaledPositionalEncoding(
27
+ (dropout): Dropout(p=0.2, inplace=False)
28
+ )
29
+ )
30
+ (encoders): MultiSequential(
31
+ (0): EncoderLayer(
32
+ (self_attn): MultiHeadedAttention(
33
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
34
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
35
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
36
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
37
+ (dropout): Dropout(p=0.2, inplace=False)
38
+ )
39
+ (feed_forward): MultiLayeredConv1d(
40
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
41
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
42
+ (dropout): Dropout(p=0.2, inplace=False)
43
+ )
44
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
45
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
46
+ (dropout): Dropout(p=0.2, inplace=False)
47
+ )
48
+ (1): EncoderLayer(
49
+ (self_attn): MultiHeadedAttention(
50
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
51
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
52
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
53
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
54
+ (dropout): Dropout(p=0.2, inplace=False)
55
+ )
56
+ (feed_forward): MultiLayeredConv1d(
57
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
58
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
59
+ (dropout): Dropout(p=0.2, inplace=False)
60
+ )
61
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
62
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
63
+ (dropout): Dropout(p=0.2, inplace=False)
64
+ )
65
+ (2): EncoderLayer(
66
+ (self_attn): MultiHeadedAttention(
67
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
68
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
69
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
70
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
71
+ (dropout): Dropout(p=0.2, inplace=False)
72
+ )
73
+ (feed_forward): MultiLayeredConv1d(
74
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
75
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
76
+ (dropout): Dropout(p=0.2, inplace=False)
77
+ )
78
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
79
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
80
+ (dropout): Dropout(p=0.2, inplace=False)
81
+ )
82
+ (3): EncoderLayer(
83
+ (self_attn): MultiHeadedAttention(
84
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
85
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
86
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
87
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
88
+ (dropout): Dropout(p=0.2, inplace=False)
89
+ )
90
+ (feed_forward): MultiLayeredConv1d(
91
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
92
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
93
+ (dropout): Dropout(p=0.2, inplace=False)
94
+ )
95
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
96
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
97
+ (dropout): Dropout(p=0.2, inplace=False)
98
+ )
99
+ )
100
+ (after_norm): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
101
+ )
102
+ (duration_predictor): DurationPredictor(
103
+ (conv): ModuleList(
104
+ (0): Sequential(
105
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
106
+ (1): ReLU()
107
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
108
+ (3): Dropout(p=0.1, inplace=False)
109
+ )
110
+ (1): Sequential(
111
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
112
+ (1): ReLU()
113
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
114
+ (3): Dropout(p=0.1, inplace=False)
115
+ )
116
+ )
117
+ (linear): Linear(in_features=256, out_features=1, bias=True)
118
+ )
119
+ (pitch_predictor): VariancePredictor(
120
+ (conv): ModuleList(
121
+ (0): Sequential(
122
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
123
+ (1): ReLU()
124
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
125
+ (3): Dropout(p=0.5, inplace=False)
126
+ )
127
+ (1): Sequential(
128
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
129
+ (1): ReLU()
130
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
131
+ (3): Dropout(p=0.5, inplace=False)
132
+ )
133
+ (2): Sequential(
134
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
135
+ (1): ReLU()
136
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
137
+ (3): Dropout(p=0.5, inplace=False)
138
+ )
139
+ (3): Sequential(
140
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
141
+ (1): ReLU()
142
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
143
+ (3): Dropout(p=0.5, inplace=False)
144
+ )
145
+ (4): Sequential(
146
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
147
+ (1): ReLU()
148
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
149
+ (3): Dropout(p=0.5, inplace=False)
150
+ )
151
+ )
152
+ (linear): Linear(in_features=256, out_features=1, bias=True)
153
+ )
154
+ (pitch_embed): Sequential(
155
+ (0): Conv1d(1, 256, kernel_size=(1,), stride=(1,))
156
+ (1): Dropout(p=0.0, inplace=False)
157
+ )
158
+ (energy_predictor): VariancePredictor(
159
+ (conv): ModuleList(
160
+ (0): Sequential(
161
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
162
+ (1): ReLU()
163
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
164
+ (3): Dropout(p=0.5, inplace=False)
165
+ )
166
+ (1): Sequential(
167
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
168
+ (1): ReLU()
169
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
170
+ (3): Dropout(p=0.5, inplace=False)
171
+ )
172
+ )
173
+ (linear): Linear(in_features=256, out_features=1, bias=True)
174
+ )
175
+ (energy_embed): Sequential(
176
+ (0): Conv1d(1, 256, kernel_size=(1,), stride=(1,))
177
+ (1): Dropout(p=0.0, inplace=False)
178
+ )
179
+ (alignment_module): AlignmentModule(
180
+ (t_conv1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
181
+ (t_conv2): Conv1d(256, 256, kernel_size=(1,), stride=(1,))
182
+ (f_conv1): Conv1d(80, 256, kernel_size=(3,), stride=(1,), padding=(1,))
183
+ (f_conv2): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
184
+ (f_conv3): Conv1d(256, 256, kernel_size=(1,), stride=(1,))
185
+ )
186
+ (length_regulator): GaussianUpsampling()
187
+ (decoder): Encoder(
188
+ (embed): Sequential(
189
+ (0): ScaledPositionalEncoding(
190
+ (dropout): Dropout(p=0.2, inplace=False)
191
+ )
192
+ )
193
+ (encoders): MultiSequential(
194
+ (0): EncoderLayer(
195
+ (self_attn): MultiHeadedAttention(
196
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
197
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
198
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
199
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
200
+ (dropout): Dropout(p=0.2, inplace=False)
201
+ )
202
+ (feed_forward): MultiLayeredConv1d(
203
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
204
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
205
+ (dropout): Dropout(p=0.2, inplace=False)
206
+ )
207
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
208
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
209
+ (dropout): Dropout(p=0.2, inplace=False)
210
+ )
211
+ (1): EncoderLayer(
212
+ (self_attn): MultiHeadedAttention(
213
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
214
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
215
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
216
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
217
+ (dropout): Dropout(p=0.2, inplace=False)
218
+ )
219
+ (feed_forward): MultiLayeredConv1d(
220
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
221
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
222
+ (dropout): Dropout(p=0.2, inplace=False)
223
+ )
224
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
225
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
226
+ (dropout): Dropout(p=0.2, inplace=False)
227
+ )
228
+ (2): EncoderLayer(
229
+ (self_attn): MultiHeadedAttention(
230
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
231
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
232
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
233
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
234
+ (dropout): Dropout(p=0.2, inplace=False)
235
+ )
236
+ (feed_forward): MultiLayeredConv1d(
237
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
238
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
239
+ (dropout): Dropout(p=0.2, inplace=False)
240
+ )
241
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
242
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
243
+ (dropout): Dropout(p=0.2, inplace=False)
244
+ )
245
+ (3): EncoderLayer(
246
+ (self_attn): MultiHeadedAttention(
247
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
248
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
249
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
250
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
251
+ (dropout): Dropout(p=0.2, inplace=False)
252
+ )
253
+ (feed_forward): MultiLayeredConv1d(
254
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
255
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
256
+ (dropout): Dropout(p=0.2, inplace=False)
257
+ )
258
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
259
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
260
+ (dropout): Dropout(p=0.2, inplace=False)
261
+ )
262
+ )
263
+ (after_norm): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
264
+ )
265
+ (generator): HiFiGANGenerator(
266
+ (input_conv): Conv1d(256, 512, kernel_size=(7,), stride=(1,), padding=(3,))
267
+ (upsamples): ModuleList(
268
+ (0): Sequential(
269
+ (0): LeakyReLU(negative_slope=0.1)
270
+ (1): ConvTranspose1d(512, 256, kernel_size=(16,), stride=(8,), padding=(4,))
271
+ )
272
+ (1): Sequential(
273
+ (0): LeakyReLU(negative_slope=0.1)
274
+ (1): ConvTranspose1d(256, 128, kernel_size=(16,), stride=(8,), padding=(4,))
275
+ )
276
+ (2): Sequential(
277
+ (0): LeakyReLU(negative_slope=0.1)
278
+ (1): ConvTranspose1d(128, 64, kernel_size=(4,), stride=(2,), padding=(1,))
279
+ )
280
+ (3): Sequential(
281
+ (0): LeakyReLU(negative_slope=0.1)
282
+ (1): ConvTranspose1d(64, 32, kernel_size=(4,), stride=(2,), padding=(1,))
283
+ )
284
+ )
285
+ (blocks): ModuleList(
286
+ (0): ResidualBlock(
287
+ (convs1): ModuleList(
288
+ (0): Sequential(
289
+ (0): LeakyReLU(negative_slope=0.1)
290
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
291
+ )
292
+ (1): Sequential(
293
+ (0): LeakyReLU(negative_slope=0.1)
294
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
295
+ )
296
+ (2): Sequential(
297
+ (0): LeakyReLU(negative_slope=0.1)
298
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
299
+ )
300
+ )
301
+ (convs2): ModuleList(
302
+ (0): Sequential(
303
+ (0): LeakyReLU(negative_slope=0.1)
304
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
305
+ )
306
+ (1): Sequential(
307
+ (0): LeakyReLU(negative_slope=0.1)
308
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
309
+ )
310
+ (2): Sequential(
311
+ (0): LeakyReLU(negative_slope=0.1)
312
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
313
+ )
314
+ )
315
+ )
316
+ (1): ResidualBlock(
317
+ (convs1): ModuleList(
318
+ (0): Sequential(
319
+ (0): LeakyReLU(negative_slope=0.1)
320
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
321
+ )
322
+ (1): Sequential(
323
+ (0): LeakyReLU(negative_slope=0.1)
324
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
325
+ )
326
+ (2): Sequential(
327
+ (0): LeakyReLU(negative_slope=0.1)
328
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
329
+ )
330
+ )
331
+ (convs2): ModuleList(
332
+ (0): Sequential(
333
+ (0): LeakyReLU(negative_slope=0.1)
334
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
335
+ )
336
+ (1): Sequential(
337
+ (0): LeakyReLU(negative_slope=0.1)
338
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
339
+ )
340
+ (2): Sequential(
341
+ (0): LeakyReLU(negative_slope=0.1)
342
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
343
+ )
344
+ )
345
+ )
346
+ (2): ResidualBlock(
347
+ (convs1): ModuleList(
348
+ (0): Sequential(
349
+ (0): LeakyReLU(negative_slope=0.1)
350
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
351
+ )
352
+ (1): Sequential(
353
+ (0): LeakyReLU(negative_slope=0.1)
354
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
355
+ )
356
+ (2): Sequential(
357
+ (0): LeakyReLU(negative_slope=0.1)
358
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
359
+ )
360
+ )
361
+ (convs2): ModuleList(
362
+ (0): Sequential(
363
+ (0): LeakyReLU(negative_slope=0.1)
364
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
365
+ )
366
+ (1): Sequential(
367
+ (0): LeakyReLU(negative_slope=0.1)
368
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
369
+ )
370
+ (2): Sequential(
371
+ (0): LeakyReLU(negative_slope=0.1)
372
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
373
+ )
374
+ )
375
+ )
376
+ (3): ResidualBlock(
377
+ (convs1): ModuleList(
378
+ (0): Sequential(
379
+ (0): LeakyReLU(negative_slope=0.1)
380
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
381
+ )
382
+ (1): Sequential(
383
+ (0): LeakyReLU(negative_slope=0.1)
384
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
385
+ )
386
+ (2): Sequential(
387
+ (0): LeakyReLU(negative_slope=0.1)
388
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
389
+ )
390
+ )
391
+ (convs2): ModuleList(
392
+ (0): Sequential(
393
+ (0): LeakyReLU(negative_slope=0.1)
394
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
395
+ )
396
+ (1): Sequential(
397
+ (0): LeakyReLU(negative_slope=0.1)
398
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
399
+ )
400
+ (2): Sequential(
401
+ (0): LeakyReLU(negative_slope=0.1)
402
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
403
+ )
404
+ )
405
+ )
406
+ (4): ResidualBlock(
407
+ (convs1): ModuleList(
408
+ (0): Sequential(
409
+ (0): LeakyReLU(negative_slope=0.1)
410
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
411
+ )
412
+ (1): Sequential(
413
+ (0): LeakyReLU(negative_slope=0.1)
414
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
415
+ )
416
+ (2): Sequential(
417
+ (0): LeakyReLU(negative_slope=0.1)
418
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
419
+ )
420
+ )
421
+ (convs2): ModuleList(
422
+ (0): Sequential(
423
+ (0): LeakyReLU(negative_slope=0.1)
424
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
425
+ )
426
+ (1): Sequential(
427
+ (0): LeakyReLU(negative_slope=0.1)
428
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
429
+ )
430
+ (2): Sequential(
431
+ (0): LeakyReLU(negative_slope=0.1)
432
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
433
+ )
434
+ )
435
+ )
436
+ (5): ResidualBlock(
437
+ (convs1): ModuleList(
438
+ (0): Sequential(
439
+ (0): LeakyReLU(negative_slope=0.1)
440
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
441
+ )
442
+ (1): Sequential(
443
+ (0): LeakyReLU(negative_slope=0.1)
444
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
445
+ )
446
+ (2): Sequential(
447
+ (0): LeakyReLU(negative_slope=0.1)
448
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
449
+ )
450
+ )
451
+ (convs2): ModuleList(
452
+ (0): Sequential(
453
+ (0): LeakyReLU(negative_slope=0.1)
454
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
455
+ )
456
+ (1): Sequential(
457
+ (0): LeakyReLU(negative_slope=0.1)
458
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
459
+ )
460
+ (2): Sequential(
461
+ (0): LeakyReLU(negative_slope=0.1)
462
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
463
+ )
464
+ )
465
+ )
466
+ (6): ResidualBlock(
467
+ (convs1): ModuleList(
468
+ (0): Sequential(
469
+ (0): LeakyReLU(negative_slope=0.1)
470
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
471
+ )
472
+ (1): Sequential(
473
+ (0): LeakyReLU(negative_slope=0.1)
474
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
475
+ )
476
+ (2): Sequential(
477
+ (0): LeakyReLU(negative_slope=0.1)
478
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
479
+ )
480
+ )
481
+ (convs2): ModuleList(
482
+ (0): Sequential(
483
+ (0): LeakyReLU(negative_slope=0.1)
484
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
485
+ )
486
+ (1): Sequential(
487
+ (0): LeakyReLU(negative_slope=0.1)
488
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
489
+ )
490
+ (2): Sequential(
491
+ (0): LeakyReLU(negative_slope=0.1)
492
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
493
+ )
494
+ )
495
+ )
496
+ (7): ResidualBlock(
497
+ (convs1): ModuleList(
498
+ (0): Sequential(
499
+ (0): LeakyReLU(negative_slope=0.1)
500
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
501
+ )
502
+ (1): Sequential(
503
+ (0): LeakyReLU(negative_slope=0.1)
504
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
505
+ )
506
+ (2): Sequential(
507
+ (0): LeakyReLU(negative_slope=0.1)
508
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
509
+ )
510
+ )
511
+ (convs2): ModuleList(
512
+ (0): Sequential(
513
+ (0): LeakyReLU(negative_slope=0.1)
514
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
515
+ )
516
+ (1): Sequential(
517
+ (0): LeakyReLU(negative_slope=0.1)
518
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
519
+ )
520
+ (2): Sequential(
521
+ (0): LeakyReLU(negative_slope=0.1)
522
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
523
+ )
524
+ )
525
+ )
526
+ (8): ResidualBlock(
527
+ (convs1): ModuleList(
528
+ (0): Sequential(
529
+ (0): LeakyReLU(negative_slope=0.1)
530
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
531
+ )
532
+ (1): Sequential(
533
+ (0): LeakyReLU(negative_slope=0.1)
534
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
535
+ )
536
+ (2): Sequential(
537
+ (0): LeakyReLU(negative_slope=0.1)
538
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
539
+ )
540
+ )
541
+ (convs2): ModuleList(
542
+ (0): Sequential(
543
+ (0): LeakyReLU(negative_slope=0.1)
544
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
545
+ )
546
+ (1): Sequential(
547
+ (0): LeakyReLU(negative_slope=0.1)
548
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
549
+ )
550
+ (2): Sequential(
551
+ (0): LeakyReLU(negative_slope=0.1)
552
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
553
+ )
554
+ )
555
+ )
556
+ (9): ResidualBlock(
557
+ (convs1): ModuleList(
558
+ (0): Sequential(
559
+ (0): LeakyReLU(negative_slope=0.1)
560
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
561
+ )
562
+ (1): Sequential(
563
+ (0): LeakyReLU(negative_slope=0.1)
564
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
565
+ )
566
+ (2): Sequential(
567
+ (0): LeakyReLU(negative_slope=0.1)
568
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
569
+ )
570
+ )
571
+ (convs2): ModuleList(
572
+ (0): Sequential(
573
+ (0): LeakyReLU(negative_slope=0.1)
574
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
575
+ )
576
+ (1): Sequential(
577
+ (0): LeakyReLU(negative_slope=0.1)
578
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
579
+ )
580
+ (2): Sequential(
581
+ (0): LeakyReLU(negative_slope=0.1)
582
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
583
+ )
584
+ )
585
+ )
586
+ (10): ResidualBlock(
587
+ (convs1): ModuleList(
588
+ (0): Sequential(
589
+ (0): LeakyReLU(negative_slope=0.1)
590
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
591
+ )
592
+ (1): Sequential(
593
+ (0): LeakyReLU(negative_slope=0.1)
594
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
595
+ )
596
+ (2): Sequential(
597
+ (0): LeakyReLU(negative_slope=0.1)
598
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
599
+ )
600
+ )
601
+ (convs2): ModuleList(
602
+ (0): Sequential(
603
+ (0): LeakyReLU(negative_slope=0.1)
604
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
605
+ )
606
+ (1): Sequential(
607
+ (0): LeakyReLU(negative_slope=0.1)
608
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
609
+ )
610
+ (2): Sequential(
611
+ (0): LeakyReLU(negative_slope=0.1)
612
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
613
+ )
614
+ )
615
+ )
616
+ (11): ResidualBlock(
617
+ (convs1): ModuleList(
618
+ (0): Sequential(
619
+ (0): LeakyReLU(negative_slope=0.1)
620
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
621
+ )
622
+ (1): Sequential(
623
+ (0): LeakyReLU(negative_slope=0.1)
624
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
625
+ )
626
+ (2): Sequential(
627
+ (0): LeakyReLU(negative_slope=0.1)
628
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
629
+ )
630
+ )
631
+ (convs2): ModuleList(
632
+ (0): Sequential(
633
+ (0): LeakyReLU(negative_slope=0.1)
634
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
635
+ )
636
+ (1): Sequential(
637
+ (0): LeakyReLU(negative_slope=0.1)
638
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
639
+ )
640
+ (2): Sequential(
641
+ (0): LeakyReLU(negative_slope=0.1)
642
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
643
+ )
644
+ )
645
+ )
646
+ )
647
+ (output_conv): Sequential(
648
+ (0): LeakyReLU(negative_slope=0.01)
649
+ (1): Conv1d(32, 1, kernel_size=(7,), stride=(1,), padding=(3,))
650
+ (2): Tanh()
651
+ )
652
+ )
653
+ )
654
+ (discriminator): HiFiGANMultiScaleMultiPeriodDiscriminator(
655
+ (msd): HiFiGANMultiScaleDiscriminator(
656
+ (discriminators): ModuleList(
657
+ (0): HiFiGANScaleDiscriminator(
658
+ (layers): ModuleList(
659
+ (0): Sequential(
660
+ (0): Conv1d(1, 128, kernel_size=(15,), stride=(1,), padding=(7,))
661
+ (1): LeakyReLU(negative_slope=0.1)
662
+ )
663
+ (1): Sequential(
664
+ (0): Conv1d(128, 128, kernel_size=(41,), stride=(2,), padding=(20,), groups=4)
665
+ (1): LeakyReLU(negative_slope=0.1)
666
+ )
667
+ (2): Sequential(
668
+ (0): Conv1d(128, 256, kernel_size=(41,), stride=(2,), padding=(20,), groups=16)
669
+ (1): LeakyReLU(negative_slope=0.1)
670
+ )
671
+ (3): Sequential(
672
+ (0): Conv1d(256, 512, kernel_size=(41,), stride=(4,), padding=(20,), groups=16)
673
+ (1): LeakyReLU(negative_slope=0.1)
674
+ )
675
+ (4): Sequential(
676
+ (0): Conv1d(512, 1024, kernel_size=(41,), stride=(4,), padding=(20,), groups=16)
677
+ (1): LeakyReLU(negative_slope=0.1)
678
+ )
679
+ (5): Sequential(
680
+ (0): Conv1d(1024, 1024, kernel_size=(41,), stride=(1,), padding=(20,), groups=16)
681
+ (1): LeakyReLU(negative_slope=0.1)
682
+ )
683
+ (6): Sequential(
684
+ (0): Conv1d(1024, 1024, kernel_size=(5,), stride=(1,), padding=(2,))
685
+ (1): LeakyReLU(negative_slope=0.1)
686
+ )
687
+ (7): Conv1d(1024, 1, kernel_size=(3,), stride=(1,), padding=(1,))
688
+ )
689
+ )
690
+ )
691
+ )
692
+ (mpd): HiFiGANMultiPeriodDiscriminator(
693
+ (discriminators): ModuleList(
694
+ (0): HiFiGANPeriodDiscriminator(
695
+ (convs): ModuleList(
696
+ (0): Sequential(
697
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
698
+ (1): LeakyReLU(negative_slope=0.1)
699
+ )
700
+ (1): Sequential(
701
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
702
+ (1): LeakyReLU(negative_slope=0.1)
703
+ )
704
+ (2): Sequential(
705
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
706
+ (1): LeakyReLU(negative_slope=0.1)
707
+ )
708
+ (3): Sequential(
709
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
710
+ (1): LeakyReLU(negative_slope=0.1)
711
+ )
712
+ (4): Sequential(
713
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
714
+ (1): LeakyReLU(negative_slope=0.1)
715
+ )
716
+ )
717
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
718
+ )
719
+ (1): HiFiGANPeriodDiscriminator(
720
+ (convs): ModuleList(
721
+ (0): Sequential(
722
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
723
+ (1): LeakyReLU(negative_slope=0.1)
724
+ )
725
+ (1): Sequential(
726
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
727
+ (1): LeakyReLU(negative_slope=0.1)
728
+ )
729
+ (2): Sequential(
730
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
731
+ (1): LeakyReLU(negative_slope=0.1)
732
+ )
733
+ (3): Sequential(
734
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
735
+ (1): LeakyReLU(negative_slope=0.1)
736
+ )
737
+ (4): Sequential(
738
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
739
+ (1): LeakyReLU(negative_slope=0.1)
740
+ )
741
+ )
742
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
743
+ )
744
+ (2): HiFiGANPeriodDiscriminator(
745
+ (convs): ModuleList(
746
+ (0): Sequential(
747
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
748
+ (1): LeakyReLU(negative_slope=0.1)
749
+ )
750
+ (1): Sequential(
751
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
752
+ (1): LeakyReLU(negative_slope=0.1)
753
+ )
754
+ (2): Sequential(
755
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
756
+ (1): LeakyReLU(negative_slope=0.1)
757
+ )
758
+ (3): Sequential(
759
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
760
+ (1): LeakyReLU(negative_slope=0.1)
761
+ )
762
+ (4): Sequential(
763
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
764
+ (1): LeakyReLU(negative_slope=0.1)
765
+ )
766
+ )
767
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
768
+ )
769
+ (3): HiFiGANPeriodDiscriminator(
770
+ (convs): ModuleList(
771
+ (0): Sequential(
772
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
773
+ (1): LeakyReLU(negative_slope=0.1)
774
+ )
775
+ (1): Sequential(
776
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
777
+ (1): LeakyReLU(negative_slope=0.1)
778
+ )
779
+ (2): Sequential(
780
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
781
+ (1): LeakyReLU(negative_slope=0.1)
782
+ )
783
+ (3): Sequential(
784
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
785
+ (1): LeakyReLU(negative_slope=0.1)
786
+ )
787
+ (4): Sequential(
788
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
789
+ (1): LeakyReLU(negative_slope=0.1)
790
+ )
791
+ )
792
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
793
+ )
794
+ (4): HiFiGANPeriodDiscriminator(
795
+ (convs): ModuleList(
796
+ (0): Sequential(
797
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
798
+ (1): LeakyReLU(negative_slope=0.1)
799
+ )
800
+ (1): Sequential(
801
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
802
+ (1): LeakyReLU(negative_slope=0.1)
803
+ )
804
+ (2): Sequential(
805
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
806
+ (1): LeakyReLU(negative_slope=0.1)
807
+ )
808
+ (3): Sequential(
809
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
810
+ (1): LeakyReLU(negative_slope=0.1)
811
+ )
812
+ (4): Sequential(
813
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
814
+ (1): LeakyReLU(negative_slope=0.1)
815
+ )
816
+ )
817
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
818
+ )
819
+ )
820
+ )
821
+ )
822
+ (generator_adv_loss): GeneratorAdversarialLoss()
823
+ (discriminator_adv_loss): DiscriminatorAdversarialLoss()
824
+ (feat_match_loss): FeatureMatchLoss()
825
+ (mel_loss): MelSpectrogramLoss(
826
+ (wav_to_mel): LogMelFbank(
827
+ (stft): Stft(n_fft=1024, win_length=1024, hop_length=256, center=True, normalized=False, onesided=True)
828
+ (logmel): LogMel(sr=24000, n_fft=1024, n_mels=80, fmin=0, fmax=12000.0, htk=False)
829
+ )
830
+ )
831
+ (var_loss): VarianceLoss(
832
+ (mse_criterion): MSELoss()
833
+ (duration_criterion): DurationPredictorLoss(
834
+ (criterion): MSELoss()
835
+ )
836
+ )
837
+ (forwardsum_loss): ForwardSumLoss()
838
+ )
839
+ )
840
+
841
+ Model summary:
842
+ Class Name: ESPnetGANTTSModel
843
+ Total Number of model parameters: 83.28 M
844
+ Number of trainable parameters: 83.28 M (100.0%)
845
+ Size: 333.11 MB
846
+ Type: torch.float32
847
+ [92b100c97f43] 2025-03-04 21:23:31,654 (abs_task:1161) INFO: Optimizer:
848
+ AdamW (
849
+ Parameter Group 0
850
+ amsgrad: False
851
+ betas: [0.8, 0.99]
852
+ eps: 1e-09
853
+ initial_lr: 0.0002
854
+ lr: 0.0002
855
+ weight_decay: 0.0
856
+ )
857
+ [92b100c97f43] 2025-03-04 21:23:31,654 (abs_task:1162) INFO: Scheduler: <torch.optim.lr_scheduler.ExponentialLR object at 0x7f31cda9c160>
858
+ [92b100c97f43] 2025-03-04 21:23:31,654 (abs_task:1161) INFO: Optimizer2:
859
+ AdamW (
860
+ Parameter Group 0
861
+ amsgrad: False
862
+ betas: [0.8, 0.99]
863
+ eps: 1e-09
864
+ initial_lr: 0.0002
865
+ lr: 0.0002
866
+ weight_decay: 0.0
867
+ )
868
+ [92b100c97f43] 2025-03-04 21:23:31,654 (abs_task:1162) INFO: Scheduler2: <torch.optim.lr_scheduler.ExponentialLR object at 0x7f31dc9c2280>
869
+ [92b100c97f43] 2025-03-04 21:23:31,654 (abs_task:1171) INFO: Saving the configuration in exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/config.yaml
870
+ [92b100c97f43] 2025-03-04 21:23:31,672 (abs_task:1182) INFO: Namespace(accum_grad=1, allow_variable_data_keys=False, batch_bins=6000000, batch_size=20, batch_type='numel', best_model_criterion=[['valid', 'text2mel_loss', 'min'], ['train', 'text2mel_loss', 'min'], ['train', 'total_count', 'max']], bpemodel=None, chunk_length=500, chunk_shift_ratio=0.5, cleaner='jaconv', collect_stats=True, config='conf/tuning/train_jets.yaml', cudnn_benchmark=False, cudnn_deterministic=False, cudnn_enabled=True, detect_anomaly=False, dist_backend='nccl', dist_init_method='env://', dist_launcher=None, dist_master_addr=None, dist_master_port=None, dist_rank=None, dist_world_size=None, distributed=False, dry_run=False, early_stopping_criterion=('valid', 'loss', 'min'), energy_extract='energy', energy_extract_conf={'reduction_factor': 1, 'use_token_averaged_energy': False, 'fs': 24000, 'n_fft': 2048, 'hop_length': 300, 'win_length': 1200}, energy_normalize=None, energy_normalize_conf={}, feats_extract='fbank', feats_extract_conf={'n_fft': 2048, 'hop_length': 300, 'win_length': 1200, 'fs': 24000, 'fmin': 80, 'fmax': 7600, 'n_mels': 80}, fold_length=[], freeze_param=[], g2p='pyopenjtalk', generator_first=True, grad_clip=-1, grad_clip_type=2.0, grad_noise=False, ignore_init_mismatch=False, init_param=[], iterator_type='sequence', keep_nbest_models=-1, local_rank=None, log_interval=50, log_level='INFO', max_cache_fd=32, max_cache_size=0.0, max_epoch=130, model_conf={}, multiple_iterator=False, multiprocessing_distributed=False, nbest_averaging_interval=0, ngpu=0, no_forward_run=False, non_linguistic_symbols=None, normalize=None, normalize_conf={}, num_att_plot=3, num_cache_chunks=1024, num_iters_per_epoch=1000, num_workers=16, odim=None, optim='adamw', optim2='adamw', optim2_conf={'lr': 0.0002, 'betas': [0.8, 0.99], 'eps': 1e-09, 'weight_decay': 0.0}, optim_conf={'lr': 0.0002, 'betas': [0.8, 0.99], 'eps': 1e-09, 'weight_decay': 0.0}, output_dir='exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2', patience=None, pitch_extract='dio', pitch_extract_conf={'reduction_factor': 1, 'use_token_averaged_f0': False, 'fs': 24000, 'n_fft': 2048, 'hop_length': 300, 'f0max': 400, 'f0min': 80}, pitch_normalize=None, pitch_normalize_conf={}, pretrain_path=None, print_config=False, required=['output_dir', 'token_list'], resume=False, scheduler='exponentiallr', scheduler2='exponentiallr', scheduler2_conf={'gamma': 0.999875}, scheduler_conf={'gamma': 0.999875}, seed=777, sharded_ddp=False, sort_batch='descending', sort_in_batch='descending', token_list=['<blank>', '<unk>', 'o', 'a', 'u', 'i', 'e', 'k', 'r', 't', 'n', 'pau', 'N', 's', 'sh', 'd', 'm', 'g', 'w', 'b', 'cl', 'I', 'j', 'ch', 'y', 'U', 'h', 'p', 'ts', 'f', 'z', 'ky', 'ny', 'gy', 'ry', 'hy', 'my', 'by', 'py', 'v', '<sos/eos>'], token_type='phn', train_data_path_and_name_and_type=[('dump/raw/jvs010_tr_no_dev/text', 'text', 'text'), ('dump/raw/jvs010_tr_no_dev/wav.scp', 'speech', 'sound')], train_dtype='float32', train_shape_file=['exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.2.scp'], tts='jets', tts_conf={'generator_type': 'jets_generator', 'generator_params': {'adim': 256, 'aheads': 2, 'elayers': 4, 'eunits': 1024, 'dlayers': 4, 'dunits': 1024, 'positionwise_layer_type': 'conv1d', 'positionwise_conv_kernel_size': 3, 'duration_predictor_layers': 2, 'duration_predictor_chans': 256, 'duration_predictor_kernel_size': 3, 'use_masking': True, 'encoder_normalize_before': True, 'decoder_normalize_before': True, 'encoder_type': 'transformer', 'decoder_type': 'transformer', 'conformer_rel_pos_type': 'latest', 'conformer_pos_enc_layer_type': 'rel_pos', 'conformer_self_attn_layer_type': 'rel_selfattn', 'conformer_activation_type': 'swish', 'use_macaron_style_in_conformer': True, 'use_cnn_in_conformer': True, 'conformer_enc_kernel_size': 7, 'conformer_dec_kernel_size': 31, 'init_type': 'xavier_uniform', 'transformer_enc_dropout_rate': 0.2, 'transformer_enc_positional_dropout_rate': 0.2, 'transformer_enc_attn_dropout_rate': 0.2, 'transformer_dec_dropout_rate': 0.2, 'transformer_dec_positional_dropout_rate': 0.2, 'transformer_dec_attn_dropout_rate': 0.2, 'pitch_predictor_layers': 5, 'pitch_predictor_chans': 256, 'pitch_predictor_kernel_size': 5, 'pitch_predictor_dropout': 0.5, 'pitch_embed_kernel_size': 1, 'pitch_embed_dropout': 0.0, 'stop_gradient_from_pitch_predictor': True, 'energy_predictor_layers': 2, 'energy_predictor_chans': 256, 'energy_predictor_kernel_size': 3, 'energy_predictor_dropout': 0.5, 'energy_embed_kernel_size': 1, 'energy_embed_dropout': 0.0, 'stop_gradient_from_energy_predictor': False, 'generator_out_channels': 1, 'generator_channels': 512, 'generator_global_channels': -1, 'generator_kernel_size': 7, 'generator_upsample_scales': [8, 8, 2, 2], 'generator_upsample_kernel_sizes': [16, 16, 4, 4], 'generator_resblock_kernel_sizes': [3, 7, 11], 'generator_resblock_dilations': [[1, 3, 5], [1, 3, 5], [1, 3, 5]], 'generator_use_additional_convs': True, 'generator_bias': True, 'generator_nonlinear_activation': 'LeakyReLU', 'generator_nonlinear_activation_params': {'negative_slope': 0.1}, 'generator_use_weight_norm': True, 'segment_size': 64, 'idim': 41, 'odim': 80}, 'discriminator_type': 'hifigan_multi_scale_multi_period_discriminator', 'discriminator_params': {'scales': 1, 'scale_downsample_pooling': 'AvgPool1d', 'scale_downsample_pooling_params': {'kernel_size': 4, 'stride': 2, 'padding': 2}, 'scale_discriminator_params': {'in_channels': 1, 'out_channels': 1, 'kernel_sizes': [15, 41, 5, 3], 'channels': 128, 'max_downsample_channels': 1024, 'max_groups': 16, 'bias': True, 'downsample_scales': [2, 2, 4, 4, 1], 'nonlinear_activation': 'LeakyReLU', 'nonlinear_activation_params': {'negative_slope': 0.1}, 'use_weight_norm': True, 'use_spectral_norm': False}, 'follow_official_norm': False, 'periods': [2, 3, 5, 7, 11], 'period_discriminator_params': {'in_channels': 1, 'out_channels': 1, 'kernel_sizes': [5, 3], 'channels': 32, 'downsample_scales': [3, 3, 3, 3, 1], 'max_downsample_channels': 1024, 'bias': True, 'nonlinear_activation': 'LeakyReLU', 'nonlinear_activation_params': {'negative_slope': 0.1}, 'use_weight_norm': True, 'use_spectral_norm': False}}, 'generator_adv_loss_params': {'average_by_discriminators': False, 'loss_type': 'mse'}, 'discriminator_adv_loss_params': {'average_by_discriminators': False, 'loss_type': 'mse'}, 'feat_match_loss_params': {'average_by_discriminators': False, 'average_by_layers': False, 'include_final_outputs': True}, 'mel_loss_params': {'fs': 24000, 'n_fft': 1024, 'hop_length': 256, 'win_length': None, 'window': 'hann', 'n_mels': 80, 'fmin': 0, 'fmax': None, 'log_base': None}, 'lambda_adv': 1.0, 'lambda_mel': 45.0, 'lambda_feat_match': 2.0, 'lambda_var': 1.0, 'lambda_align': 2.0, 'sampling_rate': 24000, 'cache_generator_outputs': True}, unused_parameters=True, use_amp=False, use_matplotlib=True, use_preprocessor=True, use_tensorboard=True, use_wandb=False, val_scheduler_criterion=('valid', 'loss'), valid_batch_bins=None, valid_batch_size=None, valid_batch_type=None, valid_data_path_and_name_and_type=[('dump/raw/jvs010_dev/text', 'text', 'text'), ('dump/raw/jvs010_dev/wav.scp', 'speech', 'sound')], valid_max_cache_size=None, valid_shape_file=['exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.2.scp'], version='202204', wandb_entity=None, wandb_id=None, wandb_model_log_interval=-1, wandb_name=None, wandb_project=None, write_collected_feats=False)
871
+ /work/espnet/espnet2/layers/stft.py:166: UserWarning: __floordiv__ is deprecated, and its behavior will change in a future version of pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). This results in incorrect rounding for negative values. To keep the current behavior, use torch.div(a, b, rounding_mode='trunc'), or for actual floor division, use torch.div(a, b, rounding_mode='floor').
872
+ olens = (ilens - self.n_fft) // self.hop_length + 1
873
+ # Accounting: time=11 threads=1
874
+ # Ended (code 0) at Tue Mar 4 21:23:37 JST 2025, elapsed time 11 seconds
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/config.yaml ADDED
@@ -0,0 +1,386 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ config: conf/tuning/train_jets.yaml
2
+ print_config: false
3
+ log_level: INFO
4
+ dry_run: false
5
+ iterator_type: sequence
6
+ output_dir: exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2
7
+ ngpu: 0
8
+ seed: 777
9
+ num_workers: 16
10
+ num_att_plot: 3
11
+ dist_backend: nccl
12
+ dist_init_method: env://
13
+ dist_world_size: null
14
+ dist_rank: null
15
+ local_rank: null
16
+ dist_master_addr: null
17
+ dist_master_port: null
18
+ dist_launcher: null
19
+ multiprocessing_distributed: false
20
+ unused_parameters: true
21
+ sharded_ddp: false
22
+ cudnn_enabled: true
23
+ cudnn_benchmark: false
24
+ cudnn_deterministic: false
25
+ collect_stats: true
26
+ write_collected_feats: false
27
+ max_epoch: 130
28
+ patience: null
29
+ val_scheduler_criterion:
30
+ - valid
31
+ - loss
32
+ early_stopping_criterion:
33
+ - valid
34
+ - loss
35
+ - min
36
+ best_model_criterion:
37
+ - - valid
38
+ - text2mel_loss
39
+ - min
40
+ - - train
41
+ - text2mel_loss
42
+ - min
43
+ - - train
44
+ - total_count
45
+ - max
46
+ keep_nbest_models: -1
47
+ nbest_averaging_interval: 0
48
+ grad_clip: -1
49
+ grad_clip_type: 2.0
50
+ grad_noise: false
51
+ accum_grad: 1
52
+ no_forward_run: false
53
+ resume: false
54
+ train_dtype: float32
55
+ use_amp: false
56
+ log_interval: 50
57
+ use_matplotlib: true
58
+ use_tensorboard: true
59
+ use_wandb: false
60
+ wandb_project: null
61
+ wandb_id: null
62
+ wandb_entity: null
63
+ wandb_name: null
64
+ wandb_model_log_interval: -1
65
+ detect_anomaly: false
66
+ pretrain_path: null
67
+ init_param: []
68
+ ignore_init_mismatch: false
69
+ freeze_param: []
70
+ num_iters_per_epoch: 1000
71
+ batch_size: 20
72
+ valid_batch_size: null
73
+ batch_bins: 6000000
74
+ valid_batch_bins: null
75
+ train_shape_file:
76
+ - exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.2.scp
77
+ valid_shape_file:
78
+ - exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.2.scp
79
+ batch_type: numel
80
+ valid_batch_type: null
81
+ fold_length: []
82
+ sort_in_batch: descending
83
+ sort_batch: descending
84
+ multiple_iterator: false
85
+ chunk_length: 500
86
+ chunk_shift_ratio: 0.5
87
+ num_cache_chunks: 1024
88
+ train_data_path_and_name_and_type:
89
+ - - dump/raw/jvs010_tr_no_dev/text
90
+ - text
91
+ - text
92
+ - - dump/raw/jvs010_tr_no_dev/wav.scp
93
+ - speech
94
+ - sound
95
+ valid_data_path_and_name_and_type:
96
+ - - dump/raw/jvs010_dev/text
97
+ - text
98
+ - text
99
+ - - dump/raw/jvs010_dev/wav.scp
100
+ - speech
101
+ - sound
102
+ allow_variable_data_keys: false
103
+ max_cache_size: 0.0
104
+ max_cache_fd: 32
105
+ valid_max_cache_size: null
106
+ optim: adamw
107
+ optim_conf:
108
+ lr: 0.0002
109
+ betas:
110
+ - 0.8
111
+ - 0.99
112
+ eps: 1.0e-09
113
+ weight_decay: 0.0
114
+ scheduler: exponentiallr
115
+ scheduler_conf:
116
+ gamma: 0.999875
117
+ optim2: adamw
118
+ optim2_conf:
119
+ lr: 0.0002
120
+ betas:
121
+ - 0.8
122
+ - 0.99
123
+ eps: 1.0e-09
124
+ weight_decay: 0.0
125
+ scheduler2: exponentiallr
126
+ scheduler2_conf:
127
+ gamma: 0.999875
128
+ generator_first: true
129
+ token_list:
130
+ - <blank>
131
+ - <unk>
132
+ - o
133
+ - a
134
+ - u
135
+ - i
136
+ - e
137
+ - k
138
+ - r
139
+ - t
140
+ - n
141
+ - pau
142
+ - N
143
+ - s
144
+ - sh
145
+ - d
146
+ - m
147
+ - g
148
+ - w
149
+ - b
150
+ - cl
151
+ - I
152
+ - j
153
+ - ch
154
+ - y
155
+ - U
156
+ - h
157
+ - p
158
+ - ts
159
+ - f
160
+ - z
161
+ - ky
162
+ - ny
163
+ - gy
164
+ - ry
165
+ - hy
166
+ - my
167
+ - by
168
+ - py
169
+ - v
170
+ - <sos/eos>
171
+ odim: null
172
+ model_conf: {}
173
+ use_preprocessor: true
174
+ token_type: phn
175
+ bpemodel: null
176
+ non_linguistic_symbols: null
177
+ cleaner: jaconv
178
+ g2p: pyopenjtalk
179
+ feats_extract: fbank
180
+ feats_extract_conf:
181
+ n_fft: 2048
182
+ hop_length: 300
183
+ win_length: 1200
184
+ fs: 24000
185
+ fmin: 80
186
+ fmax: 7600
187
+ n_mels: 80
188
+ normalize: null
189
+ normalize_conf: {}
190
+ tts: jets
191
+ tts_conf:
192
+ generator_type: jets_generator
193
+ generator_params:
194
+ adim: 256
195
+ aheads: 2
196
+ elayers: 4
197
+ eunits: 1024
198
+ dlayers: 4
199
+ dunits: 1024
200
+ positionwise_layer_type: conv1d
201
+ positionwise_conv_kernel_size: 3
202
+ duration_predictor_layers: 2
203
+ duration_predictor_chans: 256
204
+ duration_predictor_kernel_size: 3
205
+ use_masking: true
206
+ encoder_normalize_before: true
207
+ decoder_normalize_before: true
208
+ encoder_type: transformer
209
+ decoder_type: transformer
210
+ conformer_rel_pos_type: latest
211
+ conformer_pos_enc_layer_type: rel_pos
212
+ conformer_self_attn_layer_type: rel_selfattn
213
+ conformer_activation_type: swish
214
+ use_macaron_style_in_conformer: true
215
+ use_cnn_in_conformer: true
216
+ conformer_enc_kernel_size: 7
217
+ conformer_dec_kernel_size: 31
218
+ init_type: xavier_uniform
219
+ transformer_enc_dropout_rate: 0.2
220
+ transformer_enc_positional_dropout_rate: 0.2
221
+ transformer_enc_attn_dropout_rate: 0.2
222
+ transformer_dec_dropout_rate: 0.2
223
+ transformer_dec_positional_dropout_rate: 0.2
224
+ transformer_dec_attn_dropout_rate: 0.2
225
+ pitch_predictor_layers: 5
226
+ pitch_predictor_chans: 256
227
+ pitch_predictor_kernel_size: 5
228
+ pitch_predictor_dropout: 0.5
229
+ pitch_embed_kernel_size: 1
230
+ pitch_embed_dropout: 0.0
231
+ stop_gradient_from_pitch_predictor: true
232
+ energy_predictor_layers: 2
233
+ energy_predictor_chans: 256
234
+ energy_predictor_kernel_size: 3
235
+ energy_predictor_dropout: 0.5
236
+ energy_embed_kernel_size: 1
237
+ energy_embed_dropout: 0.0
238
+ stop_gradient_from_energy_predictor: false
239
+ generator_out_channels: 1
240
+ generator_channels: 512
241
+ generator_global_channels: -1
242
+ generator_kernel_size: 7
243
+ generator_upsample_scales:
244
+ - 8
245
+ - 8
246
+ - 2
247
+ - 2
248
+ generator_upsample_kernel_sizes:
249
+ - 16
250
+ - 16
251
+ - 4
252
+ - 4
253
+ generator_resblock_kernel_sizes:
254
+ - 3
255
+ - 7
256
+ - 11
257
+ generator_resblock_dilations:
258
+ - - 1
259
+ - 3
260
+ - 5
261
+ - - 1
262
+ - 3
263
+ - 5
264
+ - - 1
265
+ - 3
266
+ - 5
267
+ generator_use_additional_convs: true
268
+ generator_bias: true
269
+ generator_nonlinear_activation: LeakyReLU
270
+ generator_nonlinear_activation_params:
271
+ negative_slope: 0.1
272
+ generator_use_weight_norm: true
273
+ segment_size: 64
274
+ idim: 41
275
+ odim: 80
276
+ discriminator_type: hifigan_multi_scale_multi_period_discriminator
277
+ discriminator_params:
278
+ scales: 1
279
+ scale_downsample_pooling: AvgPool1d
280
+ scale_downsample_pooling_params:
281
+ kernel_size: 4
282
+ stride: 2
283
+ padding: 2
284
+ scale_discriminator_params:
285
+ in_channels: 1
286
+ out_channels: 1
287
+ kernel_sizes:
288
+ - 15
289
+ - 41
290
+ - 5
291
+ - 3
292
+ channels: 128
293
+ max_downsample_channels: 1024
294
+ max_groups: 16
295
+ bias: true
296
+ downsample_scales:
297
+ - 2
298
+ - 2
299
+ - 4
300
+ - 4
301
+ - 1
302
+ nonlinear_activation: LeakyReLU
303
+ nonlinear_activation_params:
304
+ negative_slope: 0.1
305
+ use_weight_norm: true
306
+ use_spectral_norm: false
307
+ follow_official_norm: false
308
+ periods:
309
+ - 2
310
+ - 3
311
+ - 5
312
+ - 7
313
+ - 11
314
+ period_discriminator_params:
315
+ in_channels: 1
316
+ out_channels: 1
317
+ kernel_sizes:
318
+ - 5
319
+ - 3
320
+ channels: 32
321
+ downsample_scales:
322
+ - 3
323
+ - 3
324
+ - 3
325
+ - 3
326
+ - 1
327
+ max_downsample_channels: 1024
328
+ bias: true
329
+ nonlinear_activation: LeakyReLU
330
+ nonlinear_activation_params:
331
+ negative_slope: 0.1
332
+ use_weight_norm: true
333
+ use_spectral_norm: false
334
+ generator_adv_loss_params:
335
+ average_by_discriminators: false
336
+ loss_type: mse
337
+ discriminator_adv_loss_params:
338
+ average_by_discriminators: false
339
+ loss_type: mse
340
+ feat_match_loss_params:
341
+ average_by_discriminators: false
342
+ average_by_layers: false
343
+ include_final_outputs: true
344
+ mel_loss_params:
345
+ fs: 24000
346
+ n_fft: 1024
347
+ hop_length: 256
348
+ win_length: null
349
+ window: hann
350
+ n_mels: 80
351
+ fmin: 0
352
+ fmax: null
353
+ log_base: null
354
+ lambda_adv: 1.0
355
+ lambda_mel: 45.0
356
+ lambda_feat_match: 2.0
357
+ lambda_var: 1.0
358
+ lambda_align: 2.0
359
+ sampling_rate: 24000
360
+ cache_generator_outputs: true
361
+ pitch_extract: dio
362
+ pitch_extract_conf:
363
+ reduction_factor: 1
364
+ use_token_averaged_f0: false
365
+ fs: 24000
366
+ n_fft: 2048
367
+ hop_length: 300
368
+ f0max: 400
369
+ f0min: 80
370
+ pitch_normalize: null
371
+ pitch_normalize_conf: {}
372
+ energy_extract: energy
373
+ energy_extract_conf:
374
+ reduction_factor: 1
375
+ use_token_averaged_energy: false
376
+ fs: 24000
377
+ n_fft: 2048
378
+ hop_length: 300
379
+ win_length: 1200
380
+ energy_normalize: null
381
+ energy_normalize_conf: {}
382
+ required:
383
+ - output_dir
384
+ - token_list
385
+ version: '202204'
386
+ distributed: false
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/batch_keys ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ text
2
+ speech
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/energy_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efe9bc4321cc000a1484565eb1b696bf9c1ed193a806ec8ebf606d4207b939d3
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/energy_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5fd01373077c58b6e8e5e20227f37180f96d8717bdf9c4e0bd17b93206fcd5a6
3
+ size 770
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/feats_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efe9bc4321cc000a1484565eb1b696bf9c1ed193a806ec8ebf606d4207b939d3
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/feats_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0205589d93485e825bd25b9c6b5e1714ca6dae977b3f555591874bacaffed623
3
+ size 1402
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/pitch_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efe9bc4321cc000a1484565eb1b696bf9c1ed193a806ec8ebf606d4207b939d3
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/pitch_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a418a92a86d6891b0a6ec4fbd84592a5df85011ea03995a231bb6e60f830f9ce
3
+ size 770
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/speech_shape ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ jvs010_VOICEACTRESS100_014 56160
2
+ jvs010_VOICEACTRESS100_015 104640
3
+ jvs010_VOICEACTRESS100_016 122880
4
+ jvs010_VOICEACTRESS100_017 148080
5
+ jvs010_VOICEACTRESS100_018 161041
6
+ jvs010_VOICEACTRESS100_019 185040
7
+ jvs010_VOICEACTRESS100_020 162000
8
+ jvs010_VOICEACTRESS100_021 154320
9
+ jvs010_VOICEACTRESS100_022 314880
10
+ jvs010_VOICEACTRESS100_023 150480
11
+ jvs010_VOICEACTRESS100_024 197040
12
+ jvs010_VOICEACTRESS100_025 98880
13
+ jvs010_VOICEACTRESS100_026 89040
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/stats_keys ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ feats
2
+ feats_lengths
3
+ pitch
4
+ pitch_lengths
5
+ energy
6
+ energy_lengths
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/train/text_shape ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ jvs010_VOICEACTRESS100_014 33
2
+ jvs010_VOICEACTRESS100_015 47
3
+ jvs010_VOICEACTRESS100_016 62
4
+ jvs010_VOICEACTRESS100_017 60
5
+ jvs010_VOICEACTRESS100_018 78
6
+ jvs010_VOICEACTRESS100_019 87
7
+ jvs010_VOICEACTRESS100_020 80
8
+ jvs010_VOICEACTRESS100_021 67
9
+ jvs010_VOICEACTRESS100_022 124
10
+ jvs010_VOICEACTRESS100_023 69
11
+ jvs010_VOICEACTRESS100_024 86
12
+ jvs010_VOICEACTRESS100_025 54
13
+ jvs010_VOICEACTRESS100_026 40
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/batch_keys ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ text
2
+ speech
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/energy_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04927f2b066ae5e9114b3f175a0890ad05a42039ba563ddf956476dde6575140
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/energy_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7a5191cf7f4b801f806ca1d7f5b73c5d530a273d73d64158189a4bf9b8c4494
3
+ size 770
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/feats_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04927f2b066ae5e9114b3f175a0890ad05a42039ba563ddf956476dde6575140
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/feats_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8766fdf1445b7fa6eaa6fffb246419dac1dac3315ee84b24a0546d58e5974aec
3
+ size 1402
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/pitch_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04927f2b066ae5e9114b3f175a0890ad05a42039ba563ddf956476dde6575140
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/pitch_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fed5ab4f8c3b3f81c11011b7f6d1eb8c0435c4c5a56361938accde0be1f3740
3
+ size 770
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/speech_shape ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ jvs010_BASIC5000_0351 72720
2
+ jvs010_BASIC5000_0882 91440
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/stats_keys ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ feats
2
+ feats_lengths
3
+ pitch
4
+ pitch_lengths
5
+ energy
6
+ energy_lengths
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.2/valid/text_shape ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ jvs010_BASIC5000_0351 36
2
+ jvs010_BASIC5000_0882 49
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3.log ADDED
@@ -0,0 +1,874 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # python3 -m espnet2.bin.gan_tts_train --collect_stats true --write_collected_feats false --use_preprocessor true --token_type phn --token_list dump/token_list/phn_jaconv_pyopenjtalk/tokens.txt --non_linguistic_symbols none --cleaner jaconv --g2p pyopenjtalk --normalize none --pitch_normalize none --energy_normalize none --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/text,text,text --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/wav.scp,speech,sound --valid_data_path_and_name_and_type dump/raw/jvs010_dev/text,text,text --valid_data_path_and_name_and_type dump/raw/jvs010_dev/wav.scp,speech,sound --train_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.3.scp --valid_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.3.scp --output_dir exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3 --config conf/tuning/train_jets.yaml --feats_extract fbank --feats_extract_conf n_fft=2048 --feats_extract_conf hop_length=300 --feats_extract_conf win_length=1200 --feats_extract_conf fs=24000 --feats_extract_conf fmin=80 --feats_extract_conf fmax=7600 --feats_extract_conf n_mels=80 --pitch_extract_conf fs=24000 --pitch_extract_conf n_fft=2048 --pitch_extract_conf hop_length=300 --pitch_extract_conf f0max=400 --pitch_extract_conf f0min=80 --energy_extract_conf fs=24000 --energy_extract_conf n_fft=2048 --energy_extract_conf hop_length=300 --energy_extract_conf win_length=1200
2
+ # Started at Tue Mar 4 21:23:26 JST 2025
3
+ #
4
+ /usr/lib/python3/dist-packages/requests/__init__.py:89: RequestsDependencyWarning: urllib3 (2.2.3) or chardet (3.0.4) doesn't match a supported version!
5
+ warnings.warn("urllib3 ({}) or chardet ({}) doesn't match a supported "
6
+ /usr/bin/python3 /work/espnet/espnet2/bin/gan_tts_train.py --collect_stats true --write_collected_feats false --use_preprocessor true --token_type phn --token_list dump/token_list/phn_jaconv_pyopenjtalk/tokens.txt --non_linguistic_symbols none --cleaner jaconv --g2p pyopenjtalk --normalize none --pitch_normalize none --energy_normalize none --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/text,text,text --train_data_path_and_name_and_type dump/raw/jvs010_tr_no_dev/wav.scp,speech,sound --valid_data_path_and_name_and_type dump/raw/jvs010_dev/text,text,text --valid_data_path_and_name_and_type dump/raw/jvs010_dev/wav.scp,speech,sound --train_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.3.scp --valid_shape_file exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.3.scp --output_dir exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3 --config conf/tuning/train_jets.yaml --feats_extract fbank --feats_extract_conf n_fft=2048 --feats_extract_conf hop_length=300 --feats_extract_conf win_length=1200 --feats_extract_conf fs=24000 --feats_extract_conf fmin=80 --feats_extract_conf fmax=7600 --feats_extract_conf n_mels=80 --pitch_extract_conf fs=24000 --pitch_extract_conf n_fft=2048 --pitch_extract_conf hop_length=300 --pitch_extract_conf f0max=400 --pitch_extract_conf f0min=80 --energy_extract_conf fs=24000 --energy_extract_conf n_fft=2048 --energy_extract_conf hop_length=300 --energy_extract_conf win_length=1200
7
+ [92b100c97f43] 2025-03-04 21:23:29,166 (gan_tts:304) INFO: Vocabulary size: 41
8
+ [92b100c97f43] 2025-03-04 21:23:29,388 (encoder:172) INFO: encoder self-attention layer type = self-attention
9
+ [92b100c97f43] 2025-03-04 21:23:29,512 (encoder:172) INFO: encoder self-attention layer type = self-attention
10
+ [92b100c97f43] 2025-03-04 21:23:31,615 (abs_task:1157) INFO: pytorch.version=1.10.1+cu113, cuda.available=True, cudnn.version=8200, cudnn.benchmark=False, cudnn.deterministic=False
11
+ [92b100c97f43] 2025-03-04 21:23:31,624 (abs_task:1158) INFO: Model structure:
12
+ ESPnetGANTTSModel(
13
+ (feats_extract): LogMelFbank(
14
+ (stft): Stft(n_fft=2048, win_length=1200, hop_length=300, center=True, normalized=False, onesided=True)
15
+ (logmel): LogMel(sr=24000, n_fft=2048, n_mels=80, fmin=80, fmax=7600, htk=False)
16
+ )
17
+ (pitch_extract): Dio()
18
+ (energy_extract): Energy(
19
+ (stft): Stft(n_fft=2048, win_length=1200, hop_length=300, center=True, normalized=False, onesided=True)
20
+ )
21
+ (tts): JETS(
22
+ (generator): JETSGenerator(
23
+ (encoder): Encoder(
24
+ (embed): Sequential(
25
+ (0): Embedding(41, 256, padding_idx=0)
26
+ (1): ScaledPositionalEncoding(
27
+ (dropout): Dropout(p=0.2, inplace=False)
28
+ )
29
+ )
30
+ (encoders): MultiSequential(
31
+ (0): EncoderLayer(
32
+ (self_attn): MultiHeadedAttention(
33
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
34
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
35
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
36
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
37
+ (dropout): Dropout(p=0.2, inplace=False)
38
+ )
39
+ (feed_forward): MultiLayeredConv1d(
40
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
41
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
42
+ (dropout): Dropout(p=0.2, inplace=False)
43
+ )
44
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
45
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
46
+ (dropout): Dropout(p=0.2, inplace=False)
47
+ )
48
+ (1): EncoderLayer(
49
+ (self_attn): MultiHeadedAttention(
50
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
51
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
52
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
53
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
54
+ (dropout): Dropout(p=0.2, inplace=False)
55
+ )
56
+ (feed_forward): MultiLayeredConv1d(
57
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
58
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
59
+ (dropout): Dropout(p=0.2, inplace=False)
60
+ )
61
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
62
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
63
+ (dropout): Dropout(p=0.2, inplace=False)
64
+ )
65
+ (2): EncoderLayer(
66
+ (self_attn): MultiHeadedAttention(
67
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
68
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
69
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
70
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
71
+ (dropout): Dropout(p=0.2, inplace=False)
72
+ )
73
+ (feed_forward): MultiLayeredConv1d(
74
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
75
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
76
+ (dropout): Dropout(p=0.2, inplace=False)
77
+ )
78
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
79
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
80
+ (dropout): Dropout(p=0.2, inplace=False)
81
+ )
82
+ (3): EncoderLayer(
83
+ (self_attn): MultiHeadedAttention(
84
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
85
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
86
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
87
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
88
+ (dropout): Dropout(p=0.2, inplace=False)
89
+ )
90
+ (feed_forward): MultiLayeredConv1d(
91
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
92
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
93
+ (dropout): Dropout(p=0.2, inplace=False)
94
+ )
95
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
96
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
97
+ (dropout): Dropout(p=0.2, inplace=False)
98
+ )
99
+ )
100
+ (after_norm): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
101
+ )
102
+ (duration_predictor): DurationPredictor(
103
+ (conv): ModuleList(
104
+ (0): Sequential(
105
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
106
+ (1): ReLU()
107
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
108
+ (3): Dropout(p=0.1, inplace=False)
109
+ )
110
+ (1): Sequential(
111
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
112
+ (1): ReLU()
113
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
114
+ (3): Dropout(p=0.1, inplace=False)
115
+ )
116
+ )
117
+ (linear): Linear(in_features=256, out_features=1, bias=True)
118
+ )
119
+ (pitch_predictor): VariancePredictor(
120
+ (conv): ModuleList(
121
+ (0): Sequential(
122
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
123
+ (1): ReLU()
124
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
125
+ (3): Dropout(p=0.5, inplace=False)
126
+ )
127
+ (1): Sequential(
128
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
129
+ (1): ReLU()
130
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
131
+ (3): Dropout(p=0.5, inplace=False)
132
+ )
133
+ (2): Sequential(
134
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
135
+ (1): ReLU()
136
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
137
+ (3): Dropout(p=0.5, inplace=False)
138
+ )
139
+ (3): Sequential(
140
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
141
+ (1): ReLU()
142
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
143
+ (3): Dropout(p=0.5, inplace=False)
144
+ )
145
+ (4): Sequential(
146
+ (0): Conv1d(256, 256, kernel_size=(5,), stride=(1,), padding=(2,))
147
+ (1): ReLU()
148
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
149
+ (3): Dropout(p=0.5, inplace=False)
150
+ )
151
+ )
152
+ (linear): Linear(in_features=256, out_features=1, bias=True)
153
+ )
154
+ (pitch_embed): Sequential(
155
+ (0): Conv1d(1, 256, kernel_size=(1,), stride=(1,))
156
+ (1): Dropout(p=0.0, inplace=False)
157
+ )
158
+ (energy_predictor): VariancePredictor(
159
+ (conv): ModuleList(
160
+ (0): Sequential(
161
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
162
+ (1): ReLU()
163
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
164
+ (3): Dropout(p=0.5, inplace=False)
165
+ )
166
+ (1): Sequential(
167
+ (0): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
168
+ (1): ReLU()
169
+ (2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
170
+ (3): Dropout(p=0.5, inplace=False)
171
+ )
172
+ )
173
+ (linear): Linear(in_features=256, out_features=1, bias=True)
174
+ )
175
+ (energy_embed): Sequential(
176
+ (0): Conv1d(1, 256, kernel_size=(1,), stride=(1,))
177
+ (1): Dropout(p=0.0, inplace=False)
178
+ )
179
+ (alignment_module): AlignmentModule(
180
+ (t_conv1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
181
+ (t_conv2): Conv1d(256, 256, kernel_size=(1,), stride=(1,))
182
+ (f_conv1): Conv1d(80, 256, kernel_size=(3,), stride=(1,), padding=(1,))
183
+ (f_conv2): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
184
+ (f_conv3): Conv1d(256, 256, kernel_size=(1,), stride=(1,))
185
+ )
186
+ (length_regulator): GaussianUpsampling()
187
+ (decoder): Encoder(
188
+ (embed): Sequential(
189
+ (0): ScaledPositionalEncoding(
190
+ (dropout): Dropout(p=0.2, inplace=False)
191
+ )
192
+ )
193
+ (encoders): MultiSequential(
194
+ (0): EncoderLayer(
195
+ (self_attn): MultiHeadedAttention(
196
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
197
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
198
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
199
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
200
+ (dropout): Dropout(p=0.2, inplace=False)
201
+ )
202
+ (feed_forward): MultiLayeredConv1d(
203
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
204
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
205
+ (dropout): Dropout(p=0.2, inplace=False)
206
+ )
207
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
208
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
209
+ (dropout): Dropout(p=0.2, inplace=False)
210
+ )
211
+ (1): EncoderLayer(
212
+ (self_attn): MultiHeadedAttention(
213
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
214
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
215
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
216
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
217
+ (dropout): Dropout(p=0.2, inplace=False)
218
+ )
219
+ (feed_forward): MultiLayeredConv1d(
220
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
221
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
222
+ (dropout): Dropout(p=0.2, inplace=False)
223
+ )
224
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
225
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
226
+ (dropout): Dropout(p=0.2, inplace=False)
227
+ )
228
+ (2): EncoderLayer(
229
+ (self_attn): MultiHeadedAttention(
230
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
231
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
232
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
233
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
234
+ (dropout): Dropout(p=0.2, inplace=False)
235
+ )
236
+ (feed_forward): MultiLayeredConv1d(
237
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
238
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
239
+ (dropout): Dropout(p=0.2, inplace=False)
240
+ )
241
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
242
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
243
+ (dropout): Dropout(p=0.2, inplace=False)
244
+ )
245
+ (3): EncoderLayer(
246
+ (self_attn): MultiHeadedAttention(
247
+ (linear_q): Linear(in_features=256, out_features=256, bias=True)
248
+ (linear_k): Linear(in_features=256, out_features=256, bias=True)
249
+ (linear_v): Linear(in_features=256, out_features=256, bias=True)
250
+ (linear_out): Linear(in_features=256, out_features=256, bias=True)
251
+ (dropout): Dropout(p=0.2, inplace=False)
252
+ )
253
+ (feed_forward): MultiLayeredConv1d(
254
+ (w_1): Conv1d(256, 1024, kernel_size=(3,), stride=(1,), padding=(1,))
255
+ (w_2): Conv1d(1024, 256, kernel_size=(3,), stride=(1,), padding=(1,))
256
+ (dropout): Dropout(p=0.2, inplace=False)
257
+ )
258
+ (norm1): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
259
+ (norm2): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
260
+ (dropout): Dropout(p=0.2, inplace=False)
261
+ )
262
+ )
263
+ (after_norm): LayerNorm((256,), eps=1e-12, elementwise_affine=True)
264
+ )
265
+ (generator): HiFiGANGenerator(
266
+ (input_conv): Conv1d(256, 512, kernel_size=(7,), stride=(1,), padding=(3,))
267
+ (upsamples): ModuleList(
268
+ (0): Sequential(
269
+ (0): LeakyReLU(negative_slope=0.1)
270
+ (1): ConvTranspose1d(512, 256, kernel_size=(16,), stride=(8,), padding=(4,))
271
+ )
272
+ (1): Sequential(
273
+ (0): LeakyReLU(negative_slope=0.1)
274
+ (1): ConvTranspose1d(256, 128, kernel_size=(16,), stride=(8,), padding=(4,))
275
+ )
276
+ (2): Sequential(
277
+ (0): LeakyReLU(negative_slope=0.1)
278
+ (1): ConvTranspose1d(128, 64, kernel_size=(4,), stride=(2,), padding=(1,))
279
+ )
280
+ (3): Sequential(
281
+ (0): LeakyReLU(negative_slope=0.1)
282
+ (1): ConvTranspose1d(64, 32, kernel_size=(4,), stride=(2,), padding=(1,))
283
+ )
284
+ )
285
+ (blocks): ModuleList(
286
+ (0): ResidualBlock(
287
+ (convs1): ModuleList(
288
+ (0): Sequential(
289
+ (0): LeakyReLU(negative_slope=0.1)
290
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
291
+ )
292
+ (1): Sequential(
293
+ (0): LeakyReLU(negative_slope=0.1)
294
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
295
+ )
296
+ (2): Sequential(
297
+ (0): LeakyReLU(negative_slope=0.1)
298
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
299
+ )
300
+ )
301
+ (convs2): ModuleList(
302
+ (0): Sequential(
303
+ (0): LeakyReLU(negative_slope=0.1)
304
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
305
+ )
306
+ (1): Sequential(
307
+ (0): LeakyReLU(negative_slope=0.1)
308
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
309
+ )
310
+ (2): Sequential(
311
+ (0): LeakyReLU(negative_slope=0.1)
312
+ (1): Conv1d(256, 256, kernel_size=(3,), stride=(1,), padding=(1,))
313
+ )
314
+ )
315
+ )
316
+ (1): ResidualBlock(
317
+ (convs1): ModuleList(
318
+ (0): Sequential(
319
+ (0): LeakyReLU(negative_slope=0.1)
320
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
321
+ )
322
+ (1): Sequential(
323
+ (0): LeakyReLU(negative_slope=0.1)
324
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
325
+ )
326
+ (2): Sequential(
327
+ (0): LeakyReLU(negative_slope=0.1)
328
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
329
+ )
330
+ )
331
+ (convs2): ModuleList(
332
+ (0): Sequential(
333
+ (0): LeakyReLU(negative_slope=0.1)
334
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
335
+ )
336
+ (1): Sequential(
337
+ (0): LeakyReLU(negative_slope=0.1)
338
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
339
+ )
340
+ (2): Sequential(
341
+ (0): LeakyReLU(negative_slope=0.1)
342
+ (1): Conv1d(256, 256, kernel_size=(7,), stride=(1,), padding=(3,))
343
+ )
344
+ )
345
+ )
346
+ (2): ResidualBlock(
347
+ (convs1): ModuleList(
348
+ (0): Sequential(
349
+ (0): LeakyReLU(negative_slope=0.1)
350
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
351
+ )
352
+ (1): Sequential(
353
+ (0): LeakyReLU(negative_slope=0.1)
354
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
355
+ )
356
+ (2): Sequential(
357
+ (0): LeakyReLU(negative_slope=0.1)
358
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
359
+ )
360
+ )
361
+ (convs2): ModuleList(
362
+ (0): Sequential(
363
+ (0): LeakyReLU(negative_slope=0.1)
364
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
365
+ )
366
+ (1): Sequential(
367
+ (0): LeakyReLU(negative_slope=0.1)
368
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
369
+ )
370
+ (2): Sequential(
371
+ (0): LeakyReLU(negative_slope=0.1)
372
+ (1): Conv1d(256, 256, kernel_size=(11,), stride=(1,), padding=(5,))
373
+ )
374
+ )
375
+ )
376
+ (3): ResidualBlock(
377
+ (convs1): ModuleList(
378
+ (0): Sequential(
379
+ (0): LeakyReLU(negative_slope=0.1)
380
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
381
+ )
382
+ (1): Sequential(
383
+ (0): LeakyReLU(negative_slope=0.1)
384
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
385
+ )
386
+ (2): Sequential(
387
+ (0): LeakyReLU(negative_slope=0.1)
388
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
389
+ )
390
+ )
391
+ (convs2): ModuleList(
392
+ (0): Sequential(
393
+ (0): LeakyReLU(negative_slope=0.1)
394
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
395
+ )
396
+ (1): Sequential(
397
+ (0): LeakyReLU(negative_slope=0.1)
398
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
399
+ )
400
+ (2): Sequential(
401
+ (0): LeakyReLU(negative_slope=0.1)
402
+ (1): Conv1d(128, 128, kernel_size=(3,), stride=(1,), padding=(1,))
403
+ )
404
+ )
405
+ )
406
+ (4): ResidualBlock(
407
+ (convs1): ModuleList(
408
+ (0): Sequential(
409
+ (0): LeakyReLU(negative_slope=0.1)
410
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
411
+ )
412
+ (1): Sequential(
413
+ (0): LeakyReLU(negative_slope=0.1)
414
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
415
+ )
416
+ (2): Sequential(
417
+ (0): LeakyReLU(negative_slope=0.1)
418
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
419
+ )
420
+ )
421
+ (convs2): ModuleList(
422
+ (0): Sequential(
423
+ (0): LeakyReLU(negative_slope=0.1)
424
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
425
+ )
426
+ (1): Sequential(
427
+ (0): LeakyReLU(negative_slope=0.1)
428
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
429
+ )
430
+ (2): Sequential(
431
+ (0): LeakyReLU(negative_slope=0.1)
432
+ (1): Conv1d(128, 128, kernel_size=(7,), stride=(1,), padding=(3,))
433
+ )
434
+ )
435
+ )
436
+ (5): ResidualBlock(
437
+ (convs1): ModuleList(
438
+ (0): Sequential(
439
+ (0): LeakyReLU(negative_slope=0.1)
440
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
441
+ )
442
+ (1): Sequential(
443
+ (0): LeakyReLU(negative_slope=0.1)
444
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
445
+ )
446
+ (2): Sequential(
447
+ (0): LeakyReLU(negative_slope=0.1)
448
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
449
+ )
450
+ )
451
+ (convs2): ModuleList(
452
+ (0): Sequential(
453
+ (0): LeakyReLU(negative_slope=0.1)
454
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
455
+ )
456
+ (1): Sequential(
457
+ (0): LeakyReLU(negative_slope=0.1)
458
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
459
+ )
460
+ (2): Sequential(
461
+ (0): LeakyReLU(negative_slope=0.1)
462
+ (1): Conv1d(128, 128, kernel_size=(11,), stride=(1,), padding=(5,))
463
+ )
464
+ )
465
+ )
466
+ (6): ResidualBlock(
467
+ (convs1): ModuleList(
468
+ (0): Sequential(
469
+ (0): LeakyReLU(negative_slope=0.1)
470
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
471
+ )
472
+ (1): Sequential(
473
+ (0): LeakyReLU(negative_slope=0.1)
474
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
475
+ )
476
+ (2): Sequential(
477
+ (0): LeakyReLU(negative_slope=0.1)
478
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
479
+ )
480
+ )
481
+ (convs2): ModuleList(
482
+ (0): Sequential(
483
+ (0): LeakyReLU(negative_slope=0.1)
484
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
485
+ )
486
+ (1): Sequential(
487
+ (0): LeakyReLU(negative_slope=0.1)
488
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
489
+ )
490
+ (2): Sequential(
491
+ (0): LeakyReLU(negative_slope=0.1)
492
+ (1): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
493
+ )
494
+ )
495
+ )
496
+ (7): ResidualBlock(
497
+ (convs1): ModuleList(
498
+ (0): Sequential(
499
+ (0): LeakyReLU(negative_slope=0.1)
500
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
501
+ )
502
+ (1): Sequential(
503
+ (0): LeakyReLU(negative_slope=0.1)
504
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
505
+ )
506
+ (2): Sequential(
507
+ (0): LeakyReLU(negative_slope=0.1)
508
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
509
+ )
510
+ )
511
+ (convs2): ModuleList(
512
+ (0): Sequential(
513
+ (0): LeakyReLU(negative_slope=0.1)
514
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
515
+ )
516
+ (1): Sequential(
517
+ (0): LeakyReLU(negative_slope=0.1)
518
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
519
+ )
520
+ (2): Sequential(
521
+ (0): LeakyReLU(negative_slope=0.1)
522
+ (1): Conv1d(64, 64, kernel_size=(7,), stride=(1,), padding=(3,))
523
+ )
524
+ )
525
+ )
526
+ (8): ResidualBlock(
527
+ (convs1): ModuleList(
528
+ (0): Sequential(
529
+ (0): LeakyReLU(negative_slope=0.1)
530
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
531
+ )
532
+ (1): Sequential(
533
+ (0): LeakyReLU(negative_slope=0.1)
534
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
535
+ )
536
+ (2): Sequential(
537
+ (0): LeakyReLU(negative_slope=0.1)
538
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
539
+ )
540
+ )
541
+ (convs2): ModuleList(
542
+ (0): Sequential(
543
+ (0): LeakyReLU(negative_slope=0.1)
544
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
545
+ )
546
+ (1): Sequential(
547
+ (0): LeakyReLU(negative_slope=0.1)
548
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
549
+ )
550
+ (2): Sequential(
551
+ (0): LeakyReLU(negative_slope=0.1)
552
+ (1): Conv1d(64, 64, kernel_size=(11,), stride=(1,), padding=(5,))
553
+ )
554
+ )
555
+ )
556
+ (9): ResidualBlock(
557
+ (convs1): ModuleList(
558
+ (0): Sequential(
559
+ (0): LeakyReLU(negative_slope=0.1)
560
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
561
+ )
562
+ (1): Sequential(
563
+ (0): LeakyReLU(negative_slope=0.1)
564
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(3,), dilation=(3,))
565
+ )
566
+ (2): Sequential(
567
+ (0): LeakyReLU(negative_slope=0.1)
568
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(5,), dilation=(5,))
569
+ )
570
+ )
571
+ (convs2): ModuleList(
572
+ (0): Sequential(
573
+ (0): LeakyReLU(negative_slope=0.1)
574
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
575
+ )
576
+ (1): Sequential(
577
+ (0): LeakyReLU(negative_slope=0.1)
578
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
579
+ )
580
+ (2): Sequential(
581
+ (0): LeakyReLU(negative_slope=0.1)
582
+ (1): Conv1d(32, 32, kernel_size=(3,), stride=(1,), padding=(1,))
583
+ )
584
+ )
585
+ )
586
+ (10): ResidualBlock(
587
+ (convs1): ModuleList(
588
+ (0): Sequential(
589
+ (0): LeakyReLU(negative_slope=0.1)
590
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
591
+ )
592
+ (1): Sequential(
593
+ (0): LeakyReLU(negative_slope=0.1)
594
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(9,), dilation=(3,))
595
+ )
596
+ (2): Sequential(
597
+ (0): LeakyReLU(negative_slope=0.1)
598
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(15,), dilation=(5,))
599
+ )
600
+ )
601
+ (convs2): ModuleList(
602
+ (0): Sequential(
603
+ (0): LeakyReLU(negative_slope=0.1)
604
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
605
+ )
606
+ (1): Sequential(
607
+ (0): LeakyReLU(negative_slope=0.1)
608
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
609
+ )
610
+ (2): Sequential(
611
+ (0): LeakyReLU(negative_slope=0.1)
612
+ (1): Conv1d(32, 32, kernel_size=(7,), stride=(1,), padding=(3,))
613
+ )
614
+ )
615
+ )
616
+ (11): ResidualBlock(
617
+ (convs1): ModuleList(
618
+ (0): Sequential(
619
+ (0): LeakyReLU(negative_slope=0.1)
620
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
621
+ )
622
+ (1): Sequential(
623
+ (0): LeakyReLU(negative_slope=0.1)
624
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(15,), dilation=(3,))
625
+ )
626
+ (2): Sequential(
627
+ (0): LeakyReLU(negative_slope=0.1)
628
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(25,), dilation=(5,))
629
+ )
630
+ )
631
+ (convs2): ModuleList(
632
+ (0): Sequential(
633
+ (0): LeakyReLU(negative_slope=0.1)
634
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
635
+ )
636
+ (1): Sequential(
637
+ (0): LeakyReLU(negative_slope=0.1)
638
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
639
+ )
640
+ (2): Sequential(
641
+ (0): LeakyReLU(negative_slope=0.1)
642
+ (1): Conv1d(32, 32, kernel_size=(11,), stride=(1,), padding=(5,))
643
+ )
644
+ )
645
+ )
646
+ )
647
+ (output_conv): Sequential(
648
+ (0): LeakyReLU(negative_slope=0.01)
649
+ (1): Conv1d(32, 1, kernel_size=(7,), stride=(1,), padding=(3,))
650
+ (2): Tanh()
651
+ )
652
+ )
653
+ )
654
+ (discriminator): HiFiGANMultiScaleMultiPeriodDiscriminator(
655
+ (msd): HiFiGANMultiScaleDiscriminator(
656
+ (discriminators): ModuleList(
657
+ (0): HiFiGANScaleDiscriminator(
658
+ (layers): ModuleList(
659
+ (0): Sequential(
660
+ (0): Conv1d(1, 128, kernel_size=(15,), stride=(1,), padding=(7,))
661
+ (1): LeakyReLU(negative_slope=0.1)
662
+ )
663
+ (1): Sequential(
664
+ (0): Conv1d(128, 128, kernel_size=(41,), stride=(2,), padding=(20,), groups=4)
665
+ (1): LeakyReLU(negative_slope=0.1)
666
+ )
667
+ (2): Sequential(
668
+ (0): Conv1d(128, 256, kernel_size=(41,), stride=(2,), padding=(20,), groups=16)
669
+ (1): LeakyReLU(negative_slope=0.1)
670
+ )
671
+ (3): Sequential(
672
+ (0): Conv1d(256, 512, kernel_size=(41,), stride=(4,), padding=(20,), groups=16)
673
+ (1): LeakyReLU(negative_slope=0.1)
674
+ )
675
+ (4): Sequential(
676
+ (0): Conv1d(512, 1024, kernel_size=(41,), stride=(4,), padding=(20,), groups=16)
677
+ (1): LeakyReLU(negative_slope=0.1)
678
+ )
679
+ (5): Sequential(
680
+ (0): Conv1d(1024, 1024, kernel_size=(41,), stride=(1,), padding=(20,), groups=16)
681
+ (1): LeakyReLU(negative_slope=0.1)
682
+ )
683
+ (6): Sequential(
684
+ (0): Conv1d(1024, 1024, kernel_size=(5,), stride=(1,), padding=(2,))
685
+ (1): LeakyReLU(negative_slope=0.1)
686
+ )
687
+ (7): Conv1d(1024, 1, kernel_size=(3,), stride=(1,), padding=(1,))
688
+ )
689
+ )
690
+ )
691
+ )
692
+ (mpd): HiFiGANMultiPeriodDiscriminator(
693
+ (discriminators): ModuleList(
694
+ (0): HiFiGANPeriodDiscriminator(
695
+ (convs): ModuleList(
696
+ (0): Sequential(
697
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
698
+ (1): LeakyReLU(negative_slope=0.1)
699
+ )
700
+ (1): Sequential(
701
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
702
+ (1): LeakyReLU(negative_slope=0.1)
703
+ )
704
+ (2): Sequential(
705
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
706
+ (1): LeakyReLU(negative_slope=0.1)
707
+ )
708
+ (3): Sequential(
709
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
710
+ (1): LeakyReLU(negative_slope=0.1)
711
+ )
712
+ (4): Sequential(
713
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
714
+ (1): LeakyReLU(negative_slope=0.1)
715
+ )
716
+ )
717
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
718
+ )
719
+ (1): HiFiGANPeriodDiscriminator(
720
+ (convs): ModuleList(
721
+ (0): Sequential(
722
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
723
+ (1): LeakyReLU(negative_slope=0.1)
724
+ )
725
+ (1): Sequential(
726
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
727
+ (1): LeakyReLU(negative_slope=0.1)
728
+ )
729
+ (2): Sequential(
730
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
731
+ (1): LeakyReLU(negative_slope=0.1)
732
+ )
733
+ (3): Sequential(
734
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
735
+ (1): LeakyReLU(negative_slope=0.1)
736
+ )
737
+ (4): Sequential(
738
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
739
+ (1): LeakyReLU(negative_slope=0.1)
740
+ )
741
+ )
742
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
743
+ )
744
+ (2): HiFiGANPeriodDiscriminator(
745
+ (convs): ModuleList(
746
+ (0): Sequential(
747
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
748
+ (1): LeakyReLU(negative_slope=0.1)
749
+ )
750
+ (1): Sequential(
751
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
752
+ (1): LeakyReLU(negative_slope=0.1)
753
+ )
754
+ (2): Sequential(
755
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
756
+ (1): LeakyReLU(negative_slope=0.1)
757
+ )
758
+ (3): Sequential(
759
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
760
+ (1): LeakyReLU(negative_slope=0.1)
761
+ )
762
+ (4): Sequential(
763
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
764
+ (1): LeakyReLU(negative_slope=0.1)
765
+ )
766
+ )
767
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
768
+ )
769
+ (3): HiFiGANPeriodDiscriminator(
770
+ (convs): ModuleList(
771
+ (0): Sequential(
772
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
773
+ (1): LeakyReLU(negative_slope=0.1)
774
+ )
775
+ (1): Sequential(
776
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
777
+ (1): LeakyReLU(negative_slope=0.1)
778
+ )
779
+ (2): Sequential(
780
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
781
+ (1): LeakyReLU(negative_slope=0.1)
782
+ )
783
+ (3): Sequential(
784
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
785
+ (1): LeakyReLU(negative_slope=0.1)
786
+ )
787
+ (4): Sequential(
788
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
789
+ (1): LeakyReLU(negative_slope=0.1)
790
+ )
791
+ )
792
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
793
+ )
794
+ (4): HiFiGANPeriodDiscriminator(
795
+ (convs): ModuleList(
796
+ (0): Sequential(
797
+ (0): Conv2d(1, 32, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
798
+ (1): LeakyReLU(negative_slope=0.1)
799
+ )
800
+ (1): Sequential(
801
+ (0): Conv2d(32, 128, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
802
+ (1): LeakyReLU(negative_slope=0.1)
803
+ )
804
+ (2): Sequential(
805
+ (0): Conv2d(128, 512, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
806
+ (1): LeakyReLU(negative_slope=0.1)
807
+ )
808
+ (3): Sequential(
809
+ (0): Conv2d(512, 1024, kernel_size=(5, 1), stride=(3, 1), padding=(2, 0))
810
+ (1): LeakyReLU(negative_slope=0.1)
811
+ )
812
+ (4): Sequential(
813
+ (0): Conv2d(1024, 1024, kernel_size=(5, 1), stride=(1, 1), padding=(2, 0))
814
+ (1): LeakyReLU(negative_slope=0.1)
815
+ )
816
+ )
817
+ (output_conv): Conv2d(1024, 1, kernel_size=(2, 1), stride=(1, 1), padding=(1, 0))
818
+ )
819
+ )
820
+ )
821
+ )
822
+ (generator_adv_loss): GeneratorAdversarialLoss()
823
+ (discriminator_adv_loss): DiscriminatorAdversarialLoss()
824
+ (feat_match_loss): FeatureMatchLoss()
825
+ (mel_loss): MelSpectrogramLoss(
826
+ (wav_to_mel): LogMelFbank(
827
+ (stft): Stft(n_fft=1024, win_length=1024, hop_length=256, center=True, normalized=False, onesided=True)
828
+ (logmel): LogMel(sr=24000, n_fft=1024, n_mels=80, fmin=0, fmax=12000.0, htk=False)
829
+ )
830
+ )
831
+ (var_loss): VarianceLoss(
832
+ (mse_criterion): MSELoss()
833
+ (duration_criterion): DurationPredictorLoss(
834
+ (criterion): MSELoss()
835
+ )
836
+ )
837
+ (forwardsum_loss): ForwardSumLoss()
838
+ )
839
+ )
840
+
841
+ Model summary:
842
+ Class Name: ESPnetGANTTSModel
843
+ Total Number of model parameters: 83.28 M
844
+ Number of trainable parameters: 83.28 M (100.0%)
845
+ Size: 333.11 MB
846
+ Type: torch.float32
847
+ [92b100c97f43] 2025-03-04 21:23:31,625 (abs_task:1161) INFO: Optimizer:
848
+ AdamW (
849
+ Parameter Group 0
850
+ amsgrad: False
851
+ betas: [0.8, 0.99]
852
+ eps: 1e-09
853
+ initial_lr: 0.0002
854
+ lr: 0.0002
855
+ weight_decay: 0.0
856
+ )
857
+ [92b100c97f43] 2025-03-04 21:23:31,625 (abs_task:1162) INFO: Scheduler: <torch.optim.lr_scheduler.ExponentialLR object at 0x7f725c8cd0a0>
858
+ [92b100c97f43] 2025-03-04 21:23:31,625 (abs_task:1161) INFO: Optimizer2:
859
+ AdamW (
860
+ Parameter Group 0
861
+ amsgrad: False
862
+ betas: [0.8, 0.99]
863
+ eps: 1e-09
864
+ initial_lr: 0.0002
865
+ lr: 0.0002
866
+ weight_decay: 0.0
867
+ )
868
+ [92b100c97f43] 2025-03-04 21:23:31,625 (abs_task:1162) INFO: Scheduler2: <torch.optim.lr_scheduler.ExponentialLR object at 0x7f7278a78280>
869
+ [92b100c97f43] 2025-03-04 21:23:31,625 (abs_task:1171) INFO: Saving the configuration in exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/config.yaml
870
+ [92b100c97f43] 2025-03-04 21:23:31,643 (abs_task:1182) INFO: Namespace(accum_grad=1, allow_variable_data_keys=False, batch_bins=6000000, batch_size=20, batch_type='numel', best_model_criterion=[['valid', 'text2mel_loss', 'min'], ['train', 'text2mel_loss', 'min'], ['train', 'total_count', 'max']], bpemodel=None, chunk_length=500, chunk_shift_ratio=0.5, cleaner='jaconv', collect_stats=True, config='conf/tuning/train_jets.yaml', cudnn_benchmark=False, cudnn_deterministic=False, cudnn_enabled=True, detect_anomaly=False, dist_backend='nccl', dist_init_method='env://', dist_launcher=None, dist_master_addr=None, dist_master_port=None, dist_rank=None, dist_world_size=None, distributed=False, dry_run=False, early_stopping_criterion=('valid', 'loss', 'min'), energy_extract='energy', energy_extract_conf={'reduction_factor': 1, 'use_token_averaged_energy': False, 'fs': 24000, 'n_fft': 2048, 'hop_length': 300, 'win_length': 1200}, energy_normalize=None, energy_normalize_conf={}, feats_extract='fbank', feats_extract_conf={'n_fft': 2048, 'hop_length': 300, 'win_length': 1200, 'fs': 24000, 'fmin': 80, 'fmax': 7600, 'n_mels': 80}, fold_length=[], freeze_param=[], g2p='pyopenjtalk', generator_first=True, grad_clip=-1, grad_clip_type=2.0, grad_noise=False, ignore_init_mismatch=False, init_param=[], iterator_type='sequence', keep_nbest_models=-1, local_rank=None, log_interval=50, log_level='INFO', max_cache_fd=32, max_cache_size=0.0, max_epoch=130, model_conf={}, multiple_iterator=False, multiprocessing_distributed=False, nbest_averaging_interval=0, ngpu=0, no_forward_run=False, non_linguistic_symbols=None, normalize=None, normalize_conf={}, num_att_plot=3, num_cache_chunks=1024, num_iters_per_epoch=1000, num_workers=16, odim=None, optim='adamw', optim2='adamw', optim2_conf={'lr': 0.0002, 'betas': [0.8, 0.99], 'eps': 1e-09, 'weight_decay': 0.0}, optim_conf={'lr': 0.0002, 'betas': [0.8, 0.99], 'eps': 1e-09, 'weight_decay': 0.0}, output_dir='exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3', patience=None, pitch_extract='dio', pitch_extract_conf={'reduction_factor': 1, 'use_token_averaged_f0': False, 'fs': 24000, 'n_fft': 2048, 'hop_length': 300, 'f0max': 400, 'f0min': 80}, pitch_normalize=None, pitch_normalize_conf={}, pretrain_path=None, print_config=False, required=['output_dir', 'token_list'], resume=False, scheduler='exponentiallr', scheduler2='exponentiallr', scheduler2_conf={'gamma': 0.999875}, scheduler_conf={'gamma': 0.999875}, seed=777, sharded_ddp=False, sort_batch='descending', sort_in_batch='descending', token_list=['<blank>', '<unk>', 'o', 'a', 'u', 'i', 'e', 'k', 'r', 't', 'n', 'pau', 'N', 's', 'sh', 'd', 'm', 'g', 'w', 'b', 'cl', 'I', 'j', 'ch', 'y', 'U', 'h', 'p', 'ts', 'f', 'z', 'ky', 'ny', 'gy', 'ry', 'hy', 'my', 'by', 'py', 'v', '<sos/eos>'], token_type='phn', train_data_path_and_name_and_type=[('dump/raw/jvs010_tr_no_dev/text', 'text', 'text'), ('dump/raw/jvs010_tr_no_dev/wav.scp', 'speech', 'sound')], train_dtype='float32', train_shape_file=['exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.3.scp'], tts='jets', tts_conf={'generator_type': 'jets_generator', 'generator_params': {'adim': 256, 'aheads': 2, 'elayers': 4, 'eunits': 1024, 'dlayers': 4, 'dunits': 1024, 'positionwise_layer_type': 'conv1d', 'positionwise_conv_kernel_size': 3, 'duration_predictor_layers': 2, 'duration_predictor_chans': 256, 'duration_predictor_kernel_size': 3, 'use_masking': True, 'encoder_normalize_before': True, 'decoder_normalize_before': True, 'encoder_type': 'transformer', 'decoder_type': 'transformer', 'conformer_rel_pos_type': 'latest', 'conformer_pos_enc_layer_type': 'rel_pos', 'conformer_self_attn_layer_type': 'rel_selfattn', 'conformer_activation_type': 'swish', 'use_macaron_style_in_conformer': True, 'use_cnn_in_conformer': True, 'conformer_enc_kernel_size': 7, 'conformer_dec_kernel_size': 31, 'init_type': 'xavier_uniform', 'transformer_enc_dropout_rate': 0.2, 'transformer_enc_positional_dropout_rate': 0.2, 'transformer_enc_attn_dropout_rate': 0.2, 'transformer_dec_dropout_rate': 0.2, 'transformer_dec_positional_dropout_rate': 0.2, 'transformer_dec_attn_dropout_rate': 0.2, 'pitch_predictor_layers': 5, 'pitch_predictor_chans': 256, 'pitch_predictor_kernel_size': 5, 'pitch_predictor_dropout': 0.5, 'pitch_embed_kernel_size': 1, 'pitch_embed_dropout': 0.0, 'stop_gradient_from_pitch_predictor': True, 'energy_predictor_layers': 2, 'energy_predictor_chans': 256, 'energy_predictor_kernel_size': 3, 'energy_predictor_dropout': 0.5, 'energy_embed_kernel_size': 1, 'energy_embed_dropout': 0.0, 'stop_gradient_from_energy_predictor': False, 'generator_out_channels': 1, 'generator_channels': 512, 'generator_global_channels': -1, 'generator_kernel_size': 7, 'generator_upsample_scales': [8, 8, 2, 2], 'generator_upsample_kernel_sizes': [16, 16, 4, 4], 'generator_resblock_kernel_sizes': [3, 7, 11], 'generator_resblock_dilations': [[1, 3, 5], [1, 3, 5], [1, 3, 5]], 'generator_use_additional_convs': True, 'generator_bias': True, 'generator_nonlinear_activation': 'LeakyReLU', 'generator_nonlinear_activation_params': {'negative_slope': 0.1}, 'generator_use_weight_norm': True, 'segment_size': 64, 'idim': 41, 'odim': 80}, 'discriminator_type': 'hifigan_multi_scale_multi_period_discriminator', 'discriminator_params': {'scales': 1, 'scale_downsample_pooling': 'AvgPool1d', 'scale_downsample_pooling_params': {'kernel_size': 4, 'stride': 2, 'padding': 2}, 'scale_discriminator_params': {'in_channels': 1, 'out_channels': 1, 'kernel_sizes': [15, 41, 5, 3], 'channels': 128, 'max_downsample_channels': 1024, 'max_groups': 16, 'bias': True, 'downsample_scales': [2, 2, 4, 4, 1], 'nonlinear_activation': 'LeakyReLU', 'nonlinear_activation_params': {'negative_slope': 0.1}, 'use_weight_norm': True, 'use_spectral_norm': False}, 'follow_official_norm': False, 'periods': [2, 3, 5, 7, 11], 'period_discriminator_params': {'in_channels': 1, 'out_channels': 1, 'kernel_sizes': [5, 3], 'channels': 32, 'downsample_scales': [3, 3, 3, 3, 1], 'max_downsample_channels': 1024, 'bias': True, 'nonlinear_activation': 'LeakyReLU', 'nonlinear_activation_params': {'negative_slope': 0.1}, 'use_weight_norm': True, 'use_spectral_norm': False}}, 'generator_adv_loss_params': {'average_by_discriminators': False, 'loss_type': 'mse'}, 'discriminator_adv_loss_params': {'average_by_discriminators': False, 'loss_type': 'mse'}, 'feat_match_loss_params': {'average_by_discriminators': False, 'average_by_layers': False, 'include_final_outputs': True}, 'mel_loss_params': {'fs': 24000, 'n_fft': 1024, 'hop_length': 256, 'win_length': None, 'window': 'hann', 'n_mels': 80, 'fmin': 0, 'fmax': None, 'log_base': None}, 'lambda_adv': 1.0, 'lambda_mel': 45.0, 'lambda_feat_match': 2.0, 'lambda_var': 1.0, 'lambda_align': 2.0, 'sampling_rate': 24000, 'cache_generator_outputs': True}, unused_parameters=True, use_amp=False, use_matplotlib=True, use_preprocessor=True, use_tensorboard=True, use_wandb=False, val_scheduler_criterion=('valid', 'loss'), valid_batch_bins=None, valid_batch_size=None, valid_batch_type=None, valid_data_path_and_name_and_type=[('dump/raw/jvs010_dev/text', 'text', 'text'), ('dump/raw/jvs010_dev/wav.scp', 'speech', 'sound')], valid_max_cache_size=None, valid_shape_file=['exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.3.scp'], version='202204', wandb_entity=None, wandb_id=None, wandb_model_log_interval=-1, wandb_name=None, wandb_project=None, write_collected_feats=False)
871
+ /work/espnet/espnet2/layers/stft.py:166: UserWarning: __floordiv__ is deprecated, and its behavior will change in a future version of pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). This results in incorrect rounding for negative values. To keep the current behavior, use torch.div(a, b, rounding_mode='trunc'), or for actual floor division, use torch.div(a, b, rounding_mode='floor').
872
+ olens = (ilens - self.n_fft) // self.hop_length + 1
873
+ # Accounting: time=11 threads=1
874
+ # Ended (code 0) at Tue Mar 4 21:23:37 JST 2025, elapsed time 11 seconds
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/config.yaml ADDED
@@ -0,0 +1,386 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ config: conf/tuning/train_jets.yaml
2
+ print_config: false
3
+ log_level: INFO
4
+ dry_run: false
5
+ iterator_type: sequence
6
+ output_dir: exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3
7
+ ngpu: 0
8
+ seed: 777
9
+ num_workers: 16
10
+ num_att_plot: 3
11
+ dist_backend: nccl
12
+ dist_init_method: env://
13
+ dist_world_size: null
14
+ dist_rank: null
15
+ local_rank: null
16
+ dist_master_addr: null
17
+ dist_master_port: null
18
+ dist_launcher: null
19
+ multiprocessing_distributed: false
20
+ unused_parameters: true
21
+ sharded_ddp: false
22
+ cudnn_enabled: true
23
+ cudnn_benchmark: false
24
+ cudnn_deterministic: false
25
+ collect_stats: true
26
+ write_collected_feats: false
27
+ max_epoch: 130
28
+ patience: null
29
+ val_scheduler_criterion:
30
+ - valid
31
+ - loss
32
+ early_stopping_criterion:
33
+ - valid
34
+ - loss
35
+ - min
36
+ best_model_criterion:
37
+ - - valid
38
+ - text2mel_loss
39
+ - min
40
+ - - train
41
+ - text2mel_loss
42
+ - min
43
+ - - train
44
+ - total_count
45
+ - max
46
+ keep_nbest_models: -1
47
+ nbest_averaging_interval: 0
48
+ grad_clip: -1
49
+ grad_clip_type: 2.0
50
+ grad_noise: false
51
+ accum_grad: 1
52
+ no_forward_run: false
53
+ resume: false
54
+ train_dtype: float32
55
+ use_amp: false
56
+ log_interval: 50
57
+ use_matplotlib: true
58
+ use_tensorboard: true
59
+ use_wandb: false
60
+ wandb_project: null
61
+ wandb_id: null
62
+ wandb_entity: null
63
+ wandb_name: null
64
+ wandb_model_log_interval: -1
65
+ detect_anomaly: false
66
+ pretrain_path: null
67
+ init_param: []
68
+ ignore_init_mismatch: false
69
+ freeze_param: []
70
+ num_iters_per_epoch: 1000
71
+ batch_size: 20
72
+ valid_batch_size: null
73
+ batch_bins: 6000000
74
+ valid_batch_bins: null
75
+ train_shape_file:
76
+ - exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/train.3.scp
77
+ valid_shape_file:
78
+ - exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/valid.3.scp
79
+ batch_type: numel
80
+ valid_batch_type: null
81
+ fold_length: []
82
+ sort_in_batch: descending
83
+ sort_batch: descending
84
+ multiple_iterator: false
85
+ chunk_length: 500
86
+ chunk_shift_ratio: 0.5
87
+ num_cache_chunks: 1024
88
+ train_data_path_and_name_and_type:
89
+ - - dump/raw/jvs010_tr_no_dev/text
90
+ - text
91
+ - text
92
+ - - dump/raw/jvs010_tr_no_dev/wav.scp
93
+ - speech
94
+ - sound
95
+ valid_data_path_and_name_and_type:
96
+ - - dump/raw/jvs010_dev/text
97
+ - text
98
+ - text
99
+ - - dump/raw/jvs010_dev/wav.scp
100
+ - speech
101
+ - sound
102
+ allow_variable_data_keys: false
103
+ max_cache_size: 0.0
104
+ max_cache_fd: 32
105
+ valid_max_cache_size: null
106
+ optim: adamw
107
+ optim_conf:
108
+ lr: 0.0002
109
+ betas:
110
+ - 0.8
111
+ - 0.99
112
+ eps: 1.0e-09
113
+ weight_decay: 0.0
114
+ scheduler: exponentiallr
115
+ scheduler_conf:
116
+ gamma: 0.999875
117
+ optim2: adamw
118
+ optim2_conf:
119
+ lr: 0.0002
120
+ betas:
121
+ - 0.8
122
+ - 0.99
123
+ eps: 1.0e-09
124
+ weight_decay: 0.0
125
+ scheduler2: exponentiallr
126
+ scheduler2_conf:
127
+ gamma: 0.999875
128
+ generator_first: true
129
+ token_list:
130
+ - <blank>
131
+ - <unk>
132
+ - o
133
+ - a
134
+ - u
135
+ - i
136
+ - e
137
+ - k
138
+ - r
139
+ - t
140
+ - n
141
+ - pau
142
+ - N
143
+ - s
144
+ - sh
145
+ - d
146
+ - m
147
+ - g
148
+ - w
149
+ - b
150
+ - cl
151
+ - I
152
+ - j
153
+ - ch
154
+ - y
155
+ - U
156
+ - h
157
+ - p
158
+ - ts
159
+ - f
160
+ - z
161
+ - ky
162
+ - ny
163
+ - gy
164
+ - ry
165
+ - hy
166
+ - my
167
+ - by
168
+ - py
169
+ - v
170
+ - <sos/eos>
171
+ odim: null
172
+ model_conf: {}
173
+ use_preprocessor: true
174
+ token_type: phn
175
+ bpemodel: null
176
+ non_linguistic_symbols: null
177
+ cleaner: jaconv
178
+ g2p: pyopenjtalk
179
+ feats_extract: fbank
180
+ feats_extract_conf:
181
+ n_fft: 2048
182
+ hop_length: 300
183
+ win_length: 1200
184
+ fs: 24000
185
+ fmin: 80
186
+ fmax: 7600
187
+ n_mels: 80
188
+ normalize: null
189
+ normalize_conf: {}
190
+ tts: jets
191
+ tts_conf:
192
+ generator_type: jets_generator
193
+ generator_params:
194
+ adim: 256
195
+ aheads: 2
196
+ elayers: 4
197
+ eunits: 1024
198
+ dlayers: 4
199
+ dunits: 1024
200
+ positionwise_layer_type: conv1d
201
+ positionwise_conv_kernel_size: 3
202
+ duration_predictor_layers: 2
203
+ duration_predictor_chans: 256
204
+ duration_predictor_kernel_size: 3
205
+ use_masking: true
206
+ encoder_normalize_before: true
207
+ decoder_normalize_before: true
208
+ encoder_type: transformer
209
+ decoder_type: transformer
210
+ conformer_rel_pos_type: latest
211
+ conformer_pos_enc_layer_type: rel_pos
212
+ conformer_self_attn_layer_type: rel_selfattn
213
+ conformer_activation_type: swish
214
+ use_macaron_style_in_conformer: true
215
+ use_cnn_in_conformer: true
216
+ conformer_enc_kernel_size: 7
217
+ conformer_dec_kernel_size: 31
218
+ init_type: xavier_uniform
219
+ transformer_enc_dropout_rate: 0.2
220
+ transformer_enc_positional_dropout_rate: 0.2
221
+ transformer_enc_attn_dropout_rate: 0.2
222
+ transformer_dec_dropout_rate: 0.2
223
+ transformer_dec_positional_dropout_rate: 0.2
224
+ transformer_dec_attn_dropout_rate: 0.2
225
+ pitch_predictor_layers: 5
226
+ pitch_predictor_chans: 256
227
+ pitch_predictor_kernel_size: 5
228
+ pitch_predictor_dropout: 0.5
229
+ pitch_embed_kernel_size: 1
230
+ pitch_embed_dropout: 0.0
231
+ stop_gradient_from_pitch_predictor: true
232
+ energy_predictor_layers: 2
233
+ energy_predictor_chans: 256
234
+ energy_predictor_kernel_size: 3
235
+ energy_predictor_dropout: 0.5
236
+ energy_embed_kernel_size: 1
237
+ energy_embed_dropout: 0.0
238
+ stop_gradient_from_energy_predictor: false
239
+ generator_out_channels: 1
240
+ generator_channels: 512
241
+ generator_global_channels: -1
242
+ generator_kernel_size: 7
243
+ generator_upsample_scales:
244
+ - 8
245
+ - 8
246
+ - 2
247
+ - 2
248
+ generator_upsample_kernel_sizes:
249
+ - 16
250
+ - 16
251
+ - 4
252
+ - 4
253
+ generator_resblock_kernel_sizes:
254
+ - 3
255
+ - 7
256
+ - 11
257
+ generator_resblock_dilations:
258
+ - - 1
259
+ - 3
260
+ - 5
261
+ - - 1
262
+ - 3
263
+ - 5
264
+ - - 1
265
+ - 3
266
+ - 5
267
+ generator_use_additional_convs: true
268
+ generator_bias: true
269
+ generator_nonlinear_activation: LeakyReLU
270
+ generator_nonlinear_activation_params:
271
+ negative_slope: 0.1
272
+ generator_use_weight_norm: true
273
+ segment_size: 64
274
+ idim: 41
275
+ odim: 80
276
+ discriminator_type: hifigan_multi_scale_multi_period_discriminator
277
+ discriminator_params:
278
+ scales: 1
279
+ scale_downsample_pooling: AvgPool1d
280
+ scale_downsample_pooling_params:
281
+ kernel_size: 4
282
+ stride: 2
283
+ padding: 2
284
+ scale_discriminator_params:
285
+ in_channels: 1
286
+ out_channels: 1
287
+ kernel_sizes:
288
+ - 15
289
+ - 41
290
+ - 5
291
+ - 3
292
+ channels: 128
293
+ max_downsample_channels: 1024
294
+ max_groups: 16
295
+ bias: true
296
+ downsample_scales:
297
+ - 2
298
+ - 2
299
+ - 4
300
+ - 4
301
+ - 1
302
+ nonlinear_activation: LeakyReLU
303
+ nonlinear_activation_params:
304
+ negative_slope: 0.1
305
+ use_weight_norm: true
306
+ use_spectral_norm: false
307
+ follow_official_norm: false
308
+ periods:
309
+ - 2
310
+ - 3
311
+ - 5
312
+ - 7
313
+ - 11
314
+ period_discriminator_params:
315
+ in_channels: 1
316
+ out_channels: 1
317
+ kernel_sizes:
318
+ - 5
319
+ - 3
320
+ channels: 32
321
+ downsample_scales:
322
+ - 3
323
+ - 3
324
+ - 3
325
+ - 3
326
+ - 1
327
+ max_downsample_channels: 1024
328
+ bias: true
329
+ nonlinear_activation: LeakyReLU
330
+ nonlinear_activation_params:
331
+ negative_slope: 0.1
332
+ use_weight_norm: true
333
+ use_spectral_norm: false
334
+ generator_adv_loss_params:
335
+ average_by_discriminators: false
336
+ loss_type: mse
337
+ discriminator_adv_loss_params:
338
+ average_by_discriminators: false
339
+ loss_type: mse
340
+ feat_match_loss_params:
341
+ average_by_discriminators: false
342
+ average_by_layers: false
343
+ include_final_outputs: true
344
+ mel_loss_params:
345
+ fs: 24000
346
+ n_fft: 1024
347
+ hop_length: 256
348
+ win_length: null
349
+ window: hann
350
+ n_mels: 80
351
+ fmin: 0
352
+ fmax: null
353
+ log_base: null
354
+ lambda_adv: 1.0
355
+ lambda_mel: 45.0
356
+ lambda_feat_match: 2.0
357
+ lambda_var: 1.0
358
+ lambda_align: 2.0
359
+ sampling_rate: 24000
360
+ cache_generator_outputs: true
361
+ pitch_extract: dio
362
+ pitch_extract_conf:
363
+ reduction_factor: 1
364
+ use_token_averaged_f0: false
365
+ fs: 24000
366
+ n_fft: 2048
367
+ hop_length: 300
368
+ f0max: 400
369
+ f0min: 80
370
+ pitch_normalize: null
371
+ pitch_normalize_conf: {}
372
+ energy_extract: energy
373
+ energy_extract_conf:
374
+ reduction_factor: 1
375
+ use_token_averaged_energy: false
376
+ fs: 24000
377
+ n_fft: 2048
378
+ hop_length: 300
379
+ win_length: 1200
380
+ energy_normalize: null
381
+ energy_normalize_conf: {}
382
+ required:
383
+ - output_dir
384
+ - token_list
385
+ version: '202204'
386
+ distributed: false
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/train/batch_keys ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ text
2
+ speech
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/train/energy_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae6a03be1e1fb8a7ead4429f0b519db24a8ba45dab0a18a87835ea8affbc2e85
3
+ size 778
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/train/energy_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ff60189b8d9d8b107ce989ba373218b4d36d74fd9e5e3b44cd70e59d9d40c97
3
+ size 770
exp/tts_stats_raw_phn_jaconv_pyopenjtalk/logdir/stats.3/train/feats_lengths_stats.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae6a03be1e1fb8a7ead4429f0b519db24a8ba45dab0a18a87835ea8affbc2e85
3
+ size 778