deepmd.pt.optimizer.KFWrapper
Module Contents
Classes
- class deepmd.pt.optimizer.KFWrapper.KFOptimizerWrapper(model: torch.nn.Module, optimizer: torch.optim.optimizer.Optimizer, atoms_selected: int, atoms_per_group: int, is_distributed: bool = False)[source]