enerzyme.tasks.optimizer

Functions

get_optimizer(name, model, hyper_params)	Get an ready-to-use optimizer for a model given the optimizer string and hyperparameters.
get_optimizer_config(**params)	Get the relevant arguments from the trainer for the optimizer name and hyperparameters, which will be used in the get_optimizer function.