From e491c935d3d309b7945c801503511a279bb09d17 Mon Sep 17 00:00:00 2001 From: knc6 Date: Sun, 14 Apr 2024 09:03:53 -0400 Subject: [PATCH] Use canonize default and pretrained with best_model.pt --- alignn/pretrained.py | 1 + 1 file changed, 1 insertion(+) diff --git a/alignn/pretrained.py b/alignn/pretrained.py index 02c7994..21c2f21 100644 --- 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))