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

Pytorch/GANs /CycleGAN/generator_model.py | Test function has a minor issue.

Open moienr opened this issue 2 years ago • 0 comments

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.

moienr avatar Oct 29 '23 17:10 moienr