Function Conv1d

usr/diff/candidate_decoder.py:29–32

(*args, **kwargs)

def Conv1d(*args, **kwargs):
    layer = nn.Conv1d(*args, **kwargs)
    nn.init.kaiming_normal_(layer.weight)
    return layer


class FFT(FastspeechDecoder):