From 5f3174e6a3bdf69f6188488495de324714a24e36 Mon Sep 17 00:00:00 2001 From: mm Date: Thu, 4 May 2023 22:00:59 +0000 Subject: [PATCH] print --- eval.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/eval.py b/eval.py index 70ad0e2..d2dc2dc 100644 --- a/eval.py +++ b/eval.py @@ -69,7 +69,8 @@ if __name__ == "__main__": data = pd.read_csv("city_distances_full.csv") # data_sample = data.sample(1_000) checkpoint_dir = "checkpoints_absmax_split" # no slash - for checkpoint in sorted(glob.glob(f"{checkpoint_dir}/*")): + for checkpoint in sorted(glob.glob(f"{checkpoint_dir}/*"))[14::]: + print(f"Evaluating {checkpoint}") data_sample = data.sample(1_000) trained_model = SentenceTransformer(checkpoint, device="cuda")