Unverified Commit b937b4d1 authored by Phil Wang's avatar Phil Wang Committed by GitHub
Browse files

Merge pull request #85 from djqualia/patch-12 

Remove print statement from debugging
parents d700bc3b 429ee43c

audiolm_pytorch/soundstream.py

+0 −1
Original line number Diff line number Diff line
@@ -145,7 +145,6 @@ class ComplexConv2d(nn.Module): 


    def forward(self, x):

        weight, bias = map(torch.view_as_complex, (self.weight, self.bias))

        print(x.dtype, weight.dtype, x.device, weight.device)

        return F.conv2d(x, weight, bias, stride = self.stride, padding = self.padding)



def ComplexSTFTResidualUnit(chan_in, chan_out, strides):