coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update docstring format

pull/792/head
Eren Gölge 2021-07-22 14:19:10 +02:00
parent b7caad39e0
commit 8584f2b82d
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/config/shared_configs.py
View File

@ -176,15 +176,20 @@ class BaseDatasetConfig(Coqpit):
Args: Args:
name (str): name (str):
Dataset name that defines the preprocessor in use. Defaults to None. Dataset name that defines the preprocessor in use. Defaults to None.
path (str): path (str):
Root path to the dataset files. Defaults to None. Root path to the dataset files. Defaults to None.
meta_file_train (str): meta_file_train (str):
Name of the dataset meta file. Or a list of speakers to be ignored at training for multi-speaker datasets. Name of the dataset meta file. Or a list of speakers to be ignored at training for multi-speaker datasets.
Defaults to None. Defaults to None.
unused_speakers (List): unused_speakers (List):
List of speakers IDs that are not used at the training. Default None. List of speakers IDs that are not used at the training. Default None.
meta_file_val (str): meta_file_val (str):
Name of the dataset meta file that defines the instances used at validation. Name of the dataset meta file that defines the instances used at validation.
meta_file_attn_mask (str): meta_file_attn_mask (str):
Path to the file that lists the attention mask files used with models that require attention masks to Path to the file that lists the attention mask files used with models that require attention masks to
train the duration predictor. train the duration predictor.

1
TTS/trainer.py
View File

@ -268,7 +268,6 @@ class Trainer:
self.config, args.restore_path, self.model, self.optimizer, self.scaler self.config, args.restore_path, self.model, self.optimizer, self.scaler
) )
# setup scheduler # setup scheduler
self.scheduler = self.get_scheduler(self.model, self.config, self.optimizer) self.scheduler = self.get_scheduler(self.model, self.config, self.optimizer)