fix(train): sub_dir_callable is not a valid kwarg to pytorch lightnin…

…g logger
EveryVoiceTTS · Sep 21, 2023 · 10fe0f5 · 10fe0f5
1 parent abc0b1c
commit 10fe0f5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/everyvoice/aligner-model.json b/everyvoice/aligner-model.json
diff --git a/everyvoice/base_cli/helpers.py b/everyvoice/base_cli/helpers.py
@@ -118,7 +118,12 @@ def train_base_command(
 
     pbar.update()
     pbar.refresh()
-    tensorboard_logger = TensorBoardLogger(**(config.training.logger.model_dump()))
+    tensorboard_logger = TensorBoardLogger(
+        **{
+            **(config.training.logger.model_dump(exclude={"sub_dir_callable": True})),
+            **{"sub_dir": config.training.logger.sub_dir},
+        }
+    )
     lr_monitor = LearningRateMonitor(logging_interval="step")
     logger.info("Starting training.")
     ckpt_callback = ModelCheckpoint(

diff --git a/everyvoice/tests/aligner-model.json b/everyvoice/tests/aligner-model.json