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:

torch.Tensor