choderalab / mtenn

Modular Training and Evaluation of Neural Networks
MIT License
5 stars 1 forks source link

Incorrect updating for GATModelConfig #64

Closed kaminow closed 2 months ago

kaminow commented 2 months ago

When updating a GATModelConfig, if a value is passed for num_layers but the original config wasn't created from num_layers, this new values is ignored. This causes issues for eg sweeps, where we might have an original model that was just built from default hyperparams but we want to sweep across different numbers of layers.