| | |
|---|
| | | |
|---|
| | | |
|---|
| | | @tables.register("frontend_classes", "DefaultFrontend") |
|---|
| | | @tables.register("frontend_classes", "EspnetFrontend") |
|---|
| | | class DefaultFrontend(nn.Module): |
|---|
| | | """Conventional frontend structure for ASR. |
|---|
| | | Stft -> WPE -> MVDR-Beamformer -> Power-spec -> Mel-Fbank -> CMVN |
|---|
| | |
|---|
| | | self, input: torch.Tensor, input_lengths: torch.Tensor |
|---|
| | | ) -> Tuple[torch.Tensor, torch.Tensor]: |
|---|
| | | # 1. Domain-conversion: e.g. Stft: time -> time-freq |
|---|
| | | # import pdb;pdb.set_trace() |
|---|
| | | if self.stft is not None: |
|---|
| | | input_stft, feats_lens = self._compute_stft(input, input_lengths) |
|---|
| | | else: |
|---|
