Fix event-free survival inference tensor use

2026-06-27 12:08:19 +08:00
parent 307704f74d
commit 00de2ff4a9
1 changed files with 1 additions and 1 deletions
--- a/evaluate_event_free_survival.py
+++ b/evaluate_event_free_survival.py
@@ -284,7 +284,7 @@ def collate_indexed_landmark_fn(batch: List[Dict[str, torch.Tensor]]) -> Dict[st
    }


-@torch.inference_mode()
+@torch.no_grad()
 def infer_landmark_hidden(
    *,
    model: DeepHealth,