VGGishBundle.VGGish¶
- class torchaudio.prototype.pipelines.VGGishBundle.VGGish¶
Implementation of VGGish model [Hershey et al., 2017].
Methods¶
forward¶
- VGGish.forward(input: Tensor) Tensor¶
- Parameters:
input (torch.Tensor) – batch of spectrograms, with shape (n_example, 1, n_frame, 64).
- Returns:
model output, with shape (n_example, 128).
- Return type:
