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")