Updated hypers for checkpoint interval in PET

metatensor · Sep 26, 2024 · 00307f6 · 00307f6
1 parent 7b68e6c
commit 00307f6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git 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
+  ENERGIES_LOSS: per_structure # per_structure or per_atom
+  CHECKPOINT_INTERVAL: 100
diff --git 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"
         }