Machine-Learning-Collection
Machine-Learning-Collection copied to clipboard
Pytorch/GANs /CycleGAN/generator_model.py | Test function has a minor issue.
Description:
In the Pytorch/GANs /CycleGAN/ generator_model.py file, there is a minor typo in the Generator class initialization. The second argument 9 refers to num_features, but it should be num_residuals instead.
This is a minor issue, but it results in a lower number of model parameters, and made me like "Wow this model is pretty small!"
Steps to Reproduce:
Open the generator_model.py file. Locate the Generator class initialization in the test function. Note that the second argument when initializing is 9.
Reproduces How Often:
Always.