Fix wrong argument name 🛠️

This commit is contained in:
Eren Gölge 2021-06-22 03:03:30 +02:00
parent 7c614525d9
commit 9790eddada
5 changed files with 5 additions and 5 deletions

View File

@ -89,7 +89,7 @@ Example run:
compute_linear_spec=False,
ap=ap,
meta_data=meta_data,
tp=C.characters if "characters" in C.keys() else None,
characters=c.characters if "characters" in C.keys() else None,
add_blank=C["add_blank"] if "add_blank" in C.keys() else False,
use_phonemes=C.use_phonemes,
phoneme_cache_path=C.phoneme_cache_path,

View File

@ -27,7 +27,7 @@ def setup_loader(ap, r, verbose=False):
compute_linear_spec=False,
meta_data=meta_data,
ap=ap,
tp=c.characters if "characters" in c.keys() else None,
characters=c.characters if "characters" in c.keys() else None,
add_blank=c["add_blank"] if "add_blank" in c.keys() else False,
batch_group_size=0,
min_seq_len=c.min_seq_len,

View File

@ -164,7 +164,7 @@ class BaseTTS(BaseModel):
compute_linear_spec=config.model.lower() == "tacotron",
meta_data=data_items,
ap=ap,
tp=config.characters,
characters=config.characters,
add_blank=config["add_blank"],
batch_group_size=0 if is_eval else config.batch_group_size * config.batch_size,
min_seq_len=config.min_seq_len,

View File

@ -112,7 +112,7 @@
"preprocessor = importlib.import_module('TTS.tts.datasets.preprocess')\n",
"preprocessor = getattr(preprocessor, DATASET.lower())\n",
"meta_data = preprocessor(DATA_PATH,METADATA_FILE)\n",
"dataset = TTSDataset(checkpoint['r'], C.text_cleaner, False, ap, meta_data,tp=C.characters if 'characters' in C.keys() else None, use_phonemes=C.use_phonemes, phoneme_cache_path=C.phoneme_cache_path, enable_eos_bos=C.enable_eos_bos_chars)\n",
"dataset = TTSDataset(checkpoint['r'], C.text_cleaner, False, ap, meta_data,characters=c.characters if 'characters' in C.keys() else None, use_phonemes=C.use_phonemes, phoneme_cache_path=C.phoneme_cache_path, enable_eos_bos=C.enable_eos_bos_chars)\n",
"loader = DataLoader(dataset, batch_size=BATCH_SIZE, num_workers=4, collate_fn=dataset.collate_fn, shuffle=False, drop_last=False)"
]
},

View File

@ -44,7 +44,7 @@ class TestTTSDataset(unittest.TestCase):
compute_linear_spec=True,
ap=self.ap,
meta_data=items,
tp=c.characters,
characters=c.characters,
batch_group_size=bgs,
min_seq_len=c.min_seq_len,
max_seq_len=float("inf"),