fix config

FAIR-Chem · Aug 8, 2024 · e5607e0 · e5607e0
1 parent 927c80f
commit e5607e0
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/configs/s2ef/all_md/equiformer_v2/finetune_oc22.yml b/configs/s2ef/all_md/equiformer_v2/finetune_oc22.yml
@@ -39,7 +39,7 @@ model:
     mode: RETAIN_BACKBONE_ONLY
     starting_checkpoint: "./checkpoints/2024-08-07-20-20-16-test/checkpoint.pt"
     heads:
-      oc22_forces:
+      forces:
         module: equiformer_v2_force_head
 
 
@@ -63,10 +63,10 @@ optim:
   loss_force: l2mae
 
 outputs:
-  # energy:
-  #   shape: 1
-  #   level: system
-  oc22_forces:
+  energy:
+    shape: 1
+    level: system
+  forces:
     irrep_dim: 1
     level: atom
     train_on_free_atoms: True