From 8584f2b82d78c385c70ad52c7ebc754817fbf3c7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Thu, 22 Jul 2021 14:19:10 +0200 Subject: [PATCH] Update docstring format --- TTS/config/shared_configs.py | 5 +++++ TTS/trainer.py | 3 +-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/TTS/config/shared_configs.py b/TTS/config/shared_configs.py index 0de3795c..d91bf2b6 100644 --- a/TTS/config/shared_configs.py +++ b/TTS/config/shared_configs.py @@ -176,15 +176,20 @@ class BaseDatasetConfig(Coqpit): Args: name (str): Dataset name that defines the preprocessor in use. Defaults to None. + path (str): Root path to the dataset files. Defaults to None. + meta_file_train (str): Name of the dataset meta file. Or a list of speakers to be ignored at training for multi-speaker datasets. Defaults to None. + unused_speakers (List): List of speakers IDs that are not used at the training. Default None. + meta_file_val (str): Name of the dataset meta file that defines the instances used at validation. + meta_file_attn_mask (str): Path to the file that lists the attention mask files used with models that require attention masks to train the duration predictor. diff --git a/TTS/trainer.py b/TTS/trainer.py index 0b4ad308..9bb5b096 100644 --- a/TTS/trainer.py +++ b/TTS/trainer.py @@ -268,8 +268,7 @@ class Trainer: 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) if self.args.continue_path: