Lightning-AI / deep-learning-project-template

Pytorch Lightning code guideline for conferences
Apache License 2.0
1.24k stars 272 forks source link

Specifying hyperparameters in LitClassifier__init__() #21

Open eladar opened 3 years ago

eladar commented 3 years ago

It is better to avoid using self.save_hyperparameters with no inputs as it will recorded backbone as part of the yaml file. Doing do with simple resnet18 take about 2 minutes (probably to serialize the model)