torchmd-net
torchmd-net copied to clipboard
Force gradient regularization
This PR introduces a new optional term to the loss that depends on the gradient of the force. The hope here is that by penalizing large force gradients the resulting forces will be smoother.