Fix 1/2 factor in energy

This commit is contained in:
rohskopf
2023-05-25 16:02:45 -06:00
parent 3012426bd5
commit 4174116014

View File

@ -15,8 +15,8 @@ def lj_potential(epsilon, sigma, rij):
lj1 = 4.0 * epsilon * sigma**12
lj2 = 4.0 * epsilon * sigma**6
eij = 0.5 * r6inv * (lj1 * r6inv - lj2)
return jnp.sum(eij), eij
eij = r6inv * (lj1 * r6inv - lj2)
return 0.5 * jnp.sum(eij), eij
# Construct a function computing _tot_e and its derivative
(_, eij), fij = jax.value_and_grad(_tot_e, has_aux=True)(rij)
return eij, fij