Hi ptrblck!
I noticed the one causing the error in the training section is this line:

        critic_real = critic(data).reshape(-1)


My input from real data is tensor of  [16, 1, 1,] and 16 is the batchsize. The one from generator has [1,1,4096]. So I just had to squeeze and unsqueeze the real data tensor to [1,1,16].
But, now I got a different error:

ValueError: Expected more than 1 spatial element when training, got input size torch.Size([1, 128, 1])

Trying to figure that out right now.