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: