Use canonize default and pretrained with best_model.pt

usnistgov · Apr 14, 2024 · e491c93 · e491c93
1 parent bbf4020
commit e491c93
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/alignn/pretrained.py b/alignn/pretrained.py
@@ -259,6 +259,7 @@ def get_figshare_model(model_name="jv_formation_energy_peratom_alignn"):
             cfg = i
         if "best_model.pt" in i:
             tmp = i
+            chks.append(i)
 
     print("Using chk file", tmp, "from ", chks)
     print("Path", os.path.abspath(path))