Train doesn't work, no response at all!

I’m learning “Dive into deep learning” book by translating mxnet code to pytorch.


I have translated GAN successfully.

https://github.com/d2l-ai/d2l-en/commits/master:


But I’m stuck almost two weeks about DCGAN

There isn’t any bugs poping up, but no response to my train.


My code: https://github.com/StevenJokes/d2l-en-read/blob/master/chapter-generative-adversarial-networks/colab_pt_DCGAN/DCGAN-trainX.ipynb
OR:
https://nbviewer.jupyter.org/github/StevenJokes/d2l-en-read/blob/master/chapter-generative-adversarial-networks/colab_pt_DCGAN/DCGAN-trainX.ipynb

Can anyone help me? Thanks in advance.
You maybe need to contrast mxnet code with my pytorch code.


Fix now.