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

Force gradient regularization

Open RaulPPelaez opened this issue 1 year ago • 0 comments

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.

RaulPPelaez avatar Jun 21 '24 07:06 RaulPPelaez