Error in "DiscriminatorSTFT"

[Liujingxiu23]

122 z = self.spec_transform(x) # [B, 2, Freq, Frames, 2]

But when I try to train the mode, z : torch.Size([8, 1, 513, 43, 2]), the second dim is 1 not 2. And errors when run z = torch.cat([z.real, z.imag], dim=1)

RuntimeError: real is not implemented for tensors with non-complex dtypes.

[Liujingxiu23]

I update torchaudio and problem solved