Open back2yes opened 6 years ago
Great job. By the way, I found this line of code doubtful:
https://github.com/znxlwm/pytorch-generative-model-collections/blob/5207b2fda3a9832c182e27dbbf2658b3078021ef/infoGAN.py#L78
Maybe this line should be changed into: a = F.sigmoid(x[:, : self.output_dim])
Agreed - just spotted your issue after making this fix in a pull request.
Great job. By the way, I found this line of code doubtful:
https://github.com/znxlwm/pytorch-generative-model-collections/blob/5207b2fda3a9832c182e27dbbf2658b3078021ef/infoGAN.py#L78
Maybe this line should be changed into: a = F.sigmoid(x[:, : self.output_dim])