no description for ECAPA2

jhauret · May 30, 2024 · 4fd6746 · 4fd6746
1 parent 5b32f8e
commit 4fd6746
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/configs/lightning_module/ecapa2.yaml b/configs/lightning_module/ecapa2.yaml
@@ -1,3 +1 @@
 _target_: vibravox.lightning_modules.ecapa2.ECAPA2LightningModule
-
-description: ${description}
diff --git a/vibravox/lightning_modules/ecapa2.py b/vibravox/lightning_modules/ecapa2.py
@@ -19,7 +19,6 @@ class ECAPA2LightningModule(LightningModule):
     def __init__(
         self,
         metrics: MetricCollection,
-        description: str = None,
     ):
         """
         Initializes the ECAPA2 model with Pytorch Lightning paradigm.
@@ -28,7 +27,6 @@ def __init__(
 
         Args:
             metrics (MetricCollection): collection of metrics to compute
-            description (str): description to log in tensorboard
         """
         super().__init__()
 
@@ -44,9 +42,6 @@ def __init__(
         # Metrics
         self.metrics = metrics
 
-        # Description string for the logger
-        self.description: str = description
-
     def training_step(self, batch):
         """
         Lightning training step: only defined for the trainer
@@ -123,7 +118,6 @@ def on_test_start(self) -> None:
         Called at the beginning of the testing loop.
 
         - Checks the LightningDataModule parameters.
-        - Logs the description in tensorboard.
         """
         # Check DataModule parameters
         self.check_datamodule_parameter()
Original file line number	Diff line number	Diff line change
		@@ -1,3 +1 @@
		_target_: vibravox.lightning_modules.ecapa2.ECAPA2LightningModule

		description: ${description}