We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 610fd8b commit 6740db2Copy full SHA for 6740db2
pyannote/audio/core/model.py
@@ -198,12 +198,13 @@ def __example_output(
198
example_output: torch.Tensor,
199
specifications: Specifications = None,
200
) -> Output:
201
- _, num_frames, dimension = example_output.shape
202
-
203
if specifications.resolution == Resolution.FRAME:
+ _, num_frames, dimension = example_output.shape
204
frame_duration = specifications.duration / num_frames
205
frames = SlidingWindow(step=frame_duration, duration=frame_duration)
206
else:
+ _, dimension = example_output.shape
207
+ num_frames = None
208
frames = None
209
210
return Output(
0 commit comments