downsize melgan test model size

This commit is contained in:
Eren Gölge 2021-06-06 13:42:52 +02:00
parent 4f29725eb6
commit 614738cc85
2 changed files with 1 additions and 2 deletions

View File

@ -112,7 +112,6 @@ def ljspeech_test(root_path, meta_file):
https://keithito.com/LJ-Speech-Dataset/"""
txt_file = os.path.join(root_path, meta_file)
items = []
speaker_name = "ljspeech"
with open(txt_file, "r", encoding="utf-8") as ttf:
for idx, line in enumerate(ttf):
cols = line.split("|")

View File

@ -19,7 +19,7 @@ config = MelganConfig(
seq_len=2048,
eval_split_size=1,
print_step=1,
discriminator_model_params={"base_channels": 16, "max_channels": 256, "downsample_factors": [4, 4, 4]},
discriminator_model_params={"base_channels": 16, "max_channels": 64, "downsample_factors": [4, 4, 4]},
print_eval=True,
data_path="tests/data/ljspeech",
output_path=output_path,