mirror of https://github.com/coqui-ai/TTS.git
Format style of the recipes
This commit is contained in:
parent
5d59100a88
commit
42862f7fdb
|
@ -7,7 +7,7 @@ def alignment_diagonal_score(alignments, binary=False):
|
||||||
binary (bool): if True, ignore scores and consider attention
|
binary (bool): if True, ignore scores and consider attention
|
||||||
as a binary mask.
|
as a binary mask.
|
||||||
Shape:
|
Shape:
|
||||||
alignments : batch x decoder_steps x encoder_steps
|
- alignments : :math:`[B, T_de, T_en]`
|
||||||
"""
|
"""
|
||||||
maxs = alignments.max(dim=1)[0]
|
maxs = alignments.max(dim=1)[0]
|
||||||
if binary:
|
if binary:
|
||||||
|
|
|
@ -693,7 +693,6 @@ class AudioProcessor(object):
|
||||||
# # plt.figure()
|
# # plt.figure()
|
||||||
# plt.imshow(spec, aspect="auto", origin="lower")
|
# plt.imshow(spec, aspect="auto", origin="lower")
|
||||||
# plt.savefig('save_img2.png')
|
# plt.savefig('save_img2.png')
|
||||||
# breakpoint()
|
|
||||||
return f0
|
return f0
|
||||||
|
|
||||||
### Audio Processing ###
|
### Audio Processing ###
|
||||||
|
|
|
@ -8,7 +8,12 @@ from TTS.utils.manage import ModelManager
|
||||||
output_path = os.path.dirname(os.path.abspath(__file__))
|
output_path = os.path.dirname(os.path.abspath(__file__))
|
||||||
|
|
||||||
# init configs
|
# init configs
|
||||||
dataset_config = BaseDatasetConfig(name="ljspeech", meta_file_train="metadata.csv", meta_file_attn_mask=os.path.join(output_path, "../LJSpeech-1.1/metadata_attn_mask.txt"), path=os.path.join(output_path, "../LJSpeech-1.1/"))
|
dataset_config = BaseDatasetConfig(
|
||||||
|
name="ljspeech",
|
||||||
|
meta_file_train="metadata.csv",
|
||||||
|
meta_file_attn_mask=os.path.join(output_path, "../LJSpeech-1.1/metadata_attn_mask.txt"),
|
||||||
|
path=os.path.join(output_path, "../LJSpeech-1.1/"),
|
||||||
|
)
|
||||||
audio_config = BaseAudioConfig(
|
audio_config = BaseAudioConfig(
|
||||||
sample_rate=22050,
|
sample_rate=22050,
|
||||||
do_trim_silence=False,
|
do_trim_silence=False,
|
||||||
|
|
Loading…
Reference in New Issue