Method _init_weights

fireredtts2/codec/decoder.py:599–602

(self, m)

        self.apply(self._init_weights)

    def _init_weights(self, m):
        if isinstance(m, nn.Conv1d):
            nn.init.trunc_normal_(m.weight, std=0.02)
            nn.init.constant_(m.bias, 0)

    def forward(self, x: torch.Tensor, x_lens: torch.Tensor):
        """