diff --git a/src/metatrain/experimental/pet/default-hypers.yaml b/src/metatrain/experimental/pet/default-hypers.yaml index 87191ab5..98839a13 100644 --- a/src/metatrain/experimental/pet/default-hypers.yaml +++ b/src/metatrain/experimental/pet/default-hypers.yaml @@ -54,4 +54,5 @@ training: DO_GRADIENT_CLIPPING: False GRADIENT_CLIPPING_MAX_NORM: null # must be overwritten if DO_GRADIENT_CLIPPING is True USE_SHIFT_AGNOSTIC_LOSS: False # only used when fitting general target. Primary use case: EDOS - ENERGIES_LOSS: per_structure # per_structure or per_atom \ No newline at end of file + ENERGIES_LOSS: per_structure # per_structure or per_atom + CHECKPOINT_INTERVAL: 100 \ No newline at end of file diff --git a/src/metatrain/experimental/pet/schema-hypers.json b/src/metatrain/experimental/pet/schema-hypers.json index 19528705..ab1ba6ff 100644 --- a/src/metatrain/experimental/pet/schema-hypers.json +++ b/src/metatrain/experimental/pet/schema-hypers.json @@ -217,6 +217,9 @@ "type": "string", "enum": ["per_structure", "per_atom"] }, + "CHECKPOINT_INTERVAL": { + "type": "integer" + }, "BALANCED_DATA_LOADER": { "type": "boolean" }