| | |
|---|
| | | # Deepcopy (In general, dict shouldn't be used as default arg) |
|---|
| | | frontend_conf = copy.deepcopy(frontend_conf) |
|---|
| | | self.hop_length = hop_length |
|---|
| | | self.fs = fs |
|---|
| | | |
|---|
| | | if apply_stft: |
|---|
| | | self.stft = Stft( |
|---|
| | |
|---|
| | | return self.n_mels |
|---|
| | | |
|---|
| | | def forward( |
|---|
| | | self, input: torch.Tensor, input_lengths: torch.Tensor |
|---|
| | | self, input: torch.Tensor, input_lengths: Union[torch.Tensor, list] |
|---|
| | | ) -> Tuple[torch.Tensor, torch.Tensor]: |
|---|
| | | if isinstance(input_lengths, list): |
|---|
| | | input_lengths = torch.tensor(input_lengths) |
|---|
| | | if input.dtype == torch.float64: |
|---|
| | | input = input.float() |
|---|
| | | # 1. Domain-conversion: e.g. Stft: time -> time-freq |
|---|
| | | if self.stft is not None: |
|---|
| | | input_stft, feats_lens = self._compute_stft(input, input_lengths) |
|---|
