torchmd-net
torchmd-net copied to clipboard
Exponential moving averages of model parameters
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.