diff --git a/main_ray_tune.py b/main_ray_tune.py index 50036dc1a665f38b9ae3041365521bff4498ceaa..85d17a7fc232b44743f6012e2af92990941a81e7 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)