The result was not as perfect as the images

[LTHsuan]

After running the code, the result was not as perfect as the images you gave.
I have tried CPU & GPU the result was till look like a mess after training 50001 epoch.
I was wonder is there any problem of the code or anything which affect the result.
I'm using the latest version you provided, and the version of pytorch was 1.8.1, the version of python was 3.7.1
Hope anyone could help

[invinceleo]

I recommend you to try another demo.