plantnet / malpolon

Deep-SDM framework
https://plantnet.github.io/malpolon/
MIT License
20 stars 5 forks source link

[Enhancement] Add optimizer & scheduler choice option in config files #55

Open tlarcher opened 2 months ago

tlarcher commented 2 months ago

Is your feature request related to a problem? Please describe. Currently it's not possible to select a specific optimizer from the yaml file. By default SGD is selected and optimizers key/values in the config files correspond to its parameters.

However, as witness with the GLC24_pre_extracted multimodal ensemble model, the optimizer needed to be changed for AdamW in the experiment's datamodule code

Describe the solution you'd like

tlarcher commented 3 days ago

Implementation in progress via #67