Unverified Commit 429ee43c authored by djqualia's avatar djqualia Committed by GitHub
Browse files

Remove print statement from debugging 

Adds a bit too much noise when printing the losses :-)
parent d700bc3b

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):