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

I have translated GAN 1 successfully.
https://github.com/d2l-ai/d2l-en/commits/master:
image1889×506 67 KB

But I’m stuck almost two weeks about DCGAN 1
There isn’t any bugs poping up, but no response to my train.
image1951×719 73.8 KB
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 1
Can anyone help me? Thanks in advance.
You maybe need to contrast mxnet 1 code with my pytorch code.