We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1853050 commit c4269e0Copy full SHA for c4269e0
mace/calculators/mace.py
@@ -315,6 +315,22 @@ def __init__(
315
for param in model.parameters():
316
param.requires_grad = False
317
318
+ def check_state(self, atoms, tol: float = 1e-15) -> list:
319
+ """
320
+ Check for any system changes since the last calculation.
321
+
322
+ Args:
323
+ atoms (ase.Atoms): The atomic structure to check.
324
+ tol (float): Tolerance for detecting changes.
325
326
+ Returns:
327
+ list: A list of changes detected in the system.
328
329
+ state = super().check_state(atoms, tol=tol)
330
+ if (not state) and (self.atoms.info != atoms.info):
331
+ state.append("info")
332
+ return state
333
334
def _create_result_tensors(
335
self, model_type: str, num_models: int, num_atoms: int
336
) -> dict:
0 commit comments