From d8f6f3236ae175feeaff455b5872d6a5fa29dca6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Muhammed=20Cihad=20ARSLANO=C4=9ELU?= Date: Wed, 21 May 2025 09:25:02 +0300 Subject: [PATCH] change mode of model from train to eval --- create_heatmaps.py | 1 + 1 file changed, 1 insertion(+) diff --git a/create_heatmaps.py b/create_heatmaps.py index e4875094..e000629d 100644 --- a/create_heatmaps.py +++ b/create_heatmaps.py @@ -33,6 +33,7 @@ device=torch.device("cuda" if torch.cuda.is_available() else "cpu") def infer_single_slide(model, features, label, reverse_label_dict, k=1): + model.eval() # The model should be in eval mode to prevent dropout and batchnorm from being applied features = features.to(device) with torch.inference_mode(): if isinstance(model, (CLAM_SB, CLAM_MB)):