Docstring update

coqui-ai · Jun 28, 2021 · 6b265ae · 6b265ae
1 parent ab563ce
commit 6b265ae
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/TTS/config/shared_configs.py b/TTS/config/shared_configs.py
@@ -180,6 +180,14 @@ class BaseTrainingConfig(Coqpit):
     among all the models.
 
     Args:
+        model (str):
+            Name of the model that is used in the training.
+        run_name (str):
+            Name of the experiment. This prefixes the output folder name.
+        run_description (str):
+            Short description of the experiment.
+        epochs (int):
+            Number training epochs. Defaults to 10000.
         batch_size (int):
             Training batch size.
         eval_batch_size (int):