torchmd-net icon indicating copy to clipboard operation
torchmd-net copied to clipboard

Exponential moving averages of model parameters

Open AntonioMirarchi opened this issue 9 months ago • 0 comments

Enhance model training by enabling Exponential Moving Average (EMA) on the model parameters. To activate this feature, simply specify the ema_prmtrs_decay arg in the configuration file. If not set, its default value is None, indicating that EMA is not going to be applied.

AntonioMirarchi avatar May 13 '24 18:05 AntonioMirarchi