JiaruiLi/DeepHealth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix event-free survival inference tensor use

Browse Source
This commit is contained in:
Jiarui Li
2026-06-27 12:08:19 +08:00
parent 307704f74d
commit 00de2ff4a9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
evaluate_event_free_survival.py
View File

@@ -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( def infer_landmark_hidden(
*, *,
model: DeepHealth, model: DeepHealth,