Fix the gradient sign for QM9q and SPICE

torchmd · Aug 26, 2022 · cbc1e04 · cbc1e04
1 parent db72e12
commit cbc1e04
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/torchmdnet/datasets/qm9q.py b/torchmdnet/datasets/qm9q.py
@@ -145,7 +145,7 @@ def sample_iter(self):
                         == "vector : Hartree/Bohr "
                     )
                     dy = (
-                        pt.tensor(mol["gradient_vector"][conf], dtype=pt.float32)
+                        -pt.tensor(mol["gradient_vector"][conf], dtype=pt.float32)
                         * self.HARTREE_TO_EV
                         / self.BORH_TO_ANGSTROM
                     )

diff --git a/torchmdnet/datasets/spice.py b/torchmdnet/datasets/spice.py
@@ -97,7 +97,7 @@ def sample_iter(self):
                 * self.HARTREE_TO_EV
             )
             all_dy = (
-                pt.tensor(mol["dft_total_gradient"], dtype=pt.float32)
+                -pt.tensor(mol["dft_total_gradient"], dtype=pt.float32)
                 * self.HARTREE_TO_EV
                 / self.BORH_TO_ANGSTROM
             )