?Typo: NetG instead of NetD?

YunzhuLi / VisGel

[CVPR 2019] Connecting Touch and Vision via Cross-Modal Prediction

http://visgel.csail.mit.edu/

68 stars 17 forks source link

?Typo: NetG instead of NetD? #1

Closed vifi2021 closed 3 years ago

vifi2021 commented 3 years ago

https://github.com/YunzhuLi/VisGel/blob/b502e37861b722168f0e52f3fb664b10fd869d0a/train.py#L192

Should "NetD" be replaced by "NetG" in this line?

YunzhuLi commented 3 years ago

Hi, thank you for your interest in our code. This line is correct, and it is supposed to be netD.

The following line compares the output with the true label. The backward function (code) will calculate the gradients for both the generator and the discriminator, and the step function will update the generator's parameters.