Merge branch 'develop' into PyanNetWavLM

pyannote · Sep 18, 2023 · 5f9211c · 5f9211c
2 parents 421ba03 + 9df6944
commit 5f9211c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pyannote/audio/core/model.py b/pyannote/audio/core/model.py
@@ -198,12 +198,13 @@ def __example_output(
             example_output: torch.Tensor,
             specifications: Specifications = None,
         ) -> Output:
-            _, num_frames, dimension = example_output.shape
-
             if specifications.resolution == Resolution.FRAME:
+                _, num_frames, dimension = example_output.shape
                 frame_duration = specifications.duration / num_frames
                 frames = SlidingWindow(step=frame_duration, duration=frame_duration)
             else:
+                _, dimension = example_output.shape
+                num_frames = None
                 frames = None
 
             return Output(

diff --git a/pyannote/audio/tasks/segmentation/multilabel.py b/pyannote/audio/tasks/segmentation/multilabel.py
@@ -277,4 +277,4 @@ def val_monitor(self):
         pytorch_lightning.callbacks.EarlyStopping
         """
 
-        return "ValLoss", "min"
+        return "loss/val", "min"