forked from MontrealCorpusTools/Montreal-Forced-Aligner
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_commandline_train.py
72 lines (66 loc) · 1.9 KB
/
test_commandline_train.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import os
import pytest
from montreal_forced_aligner.command_line.mfa import parser
from montreal_forced_aligner.command_line.train_acoustic_model import run_train_acoustic_model
from montreal_forced_aligner.g2p.trainer import G2P_DISABLED
def test_train_acoustic_with_g2p(
basic_corpus_dir,
english_us_mfa_dictionary,
generated_dir,
temp_dir,
acoustic_g2p_model_path,
):
if G2P_DISABLED:
pytest.skip("G2P disabled")
if os.path.exists(acoustic_g2p_model_path):
os.remove(acoustic_g2p_model_path)
command = [
"train",
basic_corpus_dir,
english_us_mfa_dictionary,
os.path.join(generated_dir, "basic_output"),
"-t",
temp_dir,
"--train_g2p",
"-q",
"--clean",
"--quiet",
"--debug",
"--beam",
"20",
"--retry_beam",
"80",
"-o",
acoustic_g2p_model_path,
]
args, unknown = parser.parse_known_args(command)
run_train_acoustic_model(args, unknown)
assert os.path.exists(acoustic_g2p_model_path)
def test_train_and_align_basic_speaker_dict(
multilingual_ipa_tg_corpus_dir,
mfa_speaker_dict_path,
generated_dir,
temp_dir,
basic_train_config_path,
textgrid_output_model_path,
):
if os.path.exists(textgrid_output_model_path):
os.remove(textgrid_output_model_path)
command = [
"train",
multilingual_ipa_tg_corpus_dir,
mfa_speaker_dict_path,
os.path.join(generated_dir, "ipa speaker output"),
"-t",
os.path.join(temp_dir, "temp dir with spaces"),
"--config_path",
basic_train_config_path,
"-q",
"--clean",
"--debug",
"-o",
textgrid_output_model_path,
]
args, unknown = parser.parse_known_args(command)
run_train_acoustic_model(args, unknown)
assert os.path.exists(textgrid_output_model_path)