Commit d50ae9e3 authored by Phil Wang's avatar Phil Wang
Browse files

do some improvisation and get a residual into the complex stft discriminator

parent e28ff435

audiolm_pytorch/soundstream.py

+5 −2
Original line number Diff line number Diff line
@@ -200,8 +200,11 @@ def ComplexSTFTResidualUnit(chan_in, chan_out, strides): 
    paddings = tuple(map(lambda t: t // 2, kernel_sizes))



    return nn.Sequential(

        Residual(Sequential(

            ComplexConv2d(chan_in, chan_in, 3, padding = 1),

            ModReLU(),

            ComplexConv2d(chan_in, chan_in, 3, padding = 1)

        )),

        ComplexConv2d(chan_in, chan_out, kernel_sizes, stride = strides, padding = paddings)

    )

audiolm_pytorch/version.py

+1 −1
Original line number Diff line number Diff line 
__version__ = '0.27.4'
 
__version__ = '0.28.0'