Skip to content

Commit

Permalink
Revert "epsilon 1e-6"
Browse files Browse the repository at this point in the history
This reverts commit d25a290.
  • Loading branch information
njzjz committed Jun 6, 2024
1 parent e402272 commit c03ec57
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions source/tests/pt/model/test_smooth.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ def test(
):
generator = torch.Generator(device=env.DEVICE).manual_seed(GLOBAL_SEED)
# displacement of atoms
epsilon = 1e-6 if self.epsilon is None else self.epsilon
epsilon = 1e-5 if self.epsilon is None else self.epsilon
# required prec. relative prec is not checked.
rprec = 0
aprec = 1e-5 if self.aprec is None else self.aprec
Expand Down Expand Up @@ -168,7 +168,7 @@ def setUp(self):
self.model = get_model(model_params).to(env.DEVICE)
# less degree of smoothness,
# error can be systematically removed by reducing epsilon
self.epsilon = 1e-6
self.epsilon = 1e-5
self.aprec = 1e-5


Expand All @@ -180,7 +180,7 @@ def setUp(self):
self.model = get_model(model_params).to(env.DEVICE)
# less degree of smoothness,
# error can be systematically removed by reducing epsilon
self.epsilon = 1e-6
self.epsilon = 1e-5
self.aprec = 1e-5


Expand All @@ -192,7 +192,7 @@ def setUp(self):
self.model = get_model(model_params).to(env.DEVICE)
# less degree of smoothness,
# error can be systematically removed by reducing epsilon
self.epsilon = 1e-6
self.epsilon = 1e-5
self.aprec = 1e-5


Expand All @@ -203,7 +203,7 @@ def setUp(self):
model_params["descriptor"]["repinit"]["rcut_smth"] = 3.5
self.type_split = True
self.model = get_model(model_params).to(env.DEVICE)
self.epsilon, self.aprec = 1e-6, 1e-4
self.epsilon, self.aprec = 1e-5, 1e-4


class TestEnergyModelDPA2_1(unittest.TestCase, SmoothTest):
Expand Down

0 comments on commit c03ec57

Please sign in to comment.