SHANKARMB

**I have same problem** in train Epoch: [ 2/200] [ 0/ 4] time: 101.3380, d_loss: 66.68623352, g_loss: 0.00000000 in train Epoch: [ 2/200] [ 1/ 4] time: 111.8188, d_loss: 76.59110260,...

Even I have the same error.. Even after making above changes, I get this error.. `load MNIST dataset epoch 1 Traceback (most recent call last): File "catgan_train.py", line 164, in...

Same doubt here