From 9d169877b2dc88739da6e2e3f09467afd071674a Mon Sep 17 00:00:00 2001
From: schne <leo.schneider@ecl19.ec-lyon.fr>
Date: Mon, 8 Jul 2024 16:47:42 +0200
Subject: [PATCH] fix para
---
main_ray_tune.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_ray_tune.py b/main_ray_tune.py
index 50036dc..85d17a7 100644
--- a/main_ray_tune.py
+++ b/main_ray_tune.py
@@ -204,7 +204,7 @@ def test_best_model(best_result, args):
best_trained_model.to(device)
criterion_rt = torch.nn.MSELoss()
criterion_intensity = masked_cos_sim
- checkpoint_path = os.path.join(best_result.checkpoint.to_directory(), "checkpoint.pt")
+ checkpoint_path = os.path.joi10n(best_result.checkpoint.to_directory(), "checkpoint.pt")
model_state, optimizer_state = torch.load(checkpoint_path)
best_trained_model.load_state_dict(model_state)
--
GitLab