Machine-Learning-Collection icon indicating copy to clipboard operation
Machine-Learning-Collection copied to clipboard

Published 20 hours ago • verified publisher icon aladdinpersson

pix2pix train model correction

Open vmpyr opened this issue 3 years ago • 1 comments

In train.py of the Pix2Pix research paper implementation on line 34, the code should be opt_disc.zero_grad()

The current one also seems to work, but I have a feeling that this correction would work better. Please let me know if I'm mistaken.

vmpyr avatar Jun 01 '22 06:06 vmpyr