Skip to content

Logging the model configs using save_hyperparms #302

@edyoshikun

Description

@edyoshikun

Currently only the attributes in the __init__ get recorded. If we pass the nn.Module as argument lightning save_hyperparameters() will save the whole model dictionary.

In #273 we had hard-coded some of the model parameters, but this is ugly. An alternative could be using the HyperparametersMixin
as shown here

Metadata

Metadata

Labels

documentationImprovements or additions to documentationenhancementNew feature or request

Type

No type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions