deep-learning-project-template icon indicating copy to clipboard operation
deep-learning-project-template copied to clipboard
verified publisher icon Lightning-AI

Parameter hidden_dim unused

Open cheind opened this issue 5 years ago • 0 comments

Hey,

I was just quickly browsing through the templates when I noticed that for the Autoencoder project the argparse parameter 'hidden_dim' is unused:

https://github.com/PyTorchLightning/deep-learning-project-template/blob/faef5bbf4f3c392b9e4d71e606edaab5ce4d0aaf/project/lit_autoencoder.py#L54

I suspect this should be used by the model to define the number of hidden dimensions. The other argparse parameter is probably accounted for in the Trainer (me: new to lightning). Thanks for this useful project.

cheind avatar Feb 07 '21 07:02 cheind