(self, n_mel_bins=80, conv_layers=2)
| 118 | |
| 119 | class PitchExtractor(nn.Module): |
| 120 | def __init__(self, n_mel_bins=80, conv_layers=2): |
| 121 | super().__init__() |
| 122 | self.hidden_size = hparams['hidden_size'] |
| 123 | self.predictor_hidden = hparams['predictor_hidden'] if hparams['predictor_hidden'] > 0 else self.hidden_size |
| 124 | self.conv_layers = conv_layers |
| 125 | |
| 126 | self.mel_prenet = Prenet(n_mel_bins, self.hidden_size, strides=[1, 1, 1]) |
| 127 | if self.conv_layers > 0: |
| 128 | self.mel_encoder = ConvStacks( |
| 129 | idim=self.hidden_size, n_chans=self.hidden_size, odim=self.hidden_size, n_layers=self.conv_layers) |
| 130 | self.pitch_predictor = PitchPredictor( |
| 131 | self.hidden_size, n_chans=self.predictor_hidden, |
| 132 | n_layers=5, dropout_rate=0.1, odim=2, |
| 133 | padding=hparams['ffn_padding'], kernel_size=hparams['predictor_kernel']) |
| 134 | |
| 135 | def forward(self, mel_input=None): |
| 136 | ret = {} |
nothing calls this directly
no test coverage detected