Update default hypers

metatensor · Sep 24, 2024 · 7c5c22d · 7c5c22d
1 parent de08f9f
commit 7c5c22d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/metatrain/experimental/pet/default-hypers.yaml b/src/metatrain/experimental/pet/default-hypers.yaml
@@ -37,9 +37,7 @@ training:
   INITIAL_LR: 1e-4
   EPOCH_NUM: 1000
   EPOCHS_WARMUP: 50
-  EPOCH_NUM_ATOMIC: 1000000000000000000 # structural version is called "EPOCH_NUM"
   SCHEDULER_STEP_SIZE_ATOMIC: 500000000 # structural version is called "SCHEDULER_STEP_SIZE"
-  EPOCHS_WARMUP_ATOMIC: 250000000 # structural version is called "EPOCHS_WARMUP"
   GLOBAL_AUG: True
   SLIDING_FACTOR: 0.7
   ATOMIC_BATCH_SIZE: 850 # structural version is called "STRUCTURAL_BATCH_SIZE"