From e1b301fff30127d5088280cdfe9e27b0a099d398 Mon Sep 17 00:00:00 2001
From: Joan Gibert <33808607+Tato14@users.noreply.github.com>
Date: Thu, 24 Feb 2022 15:27:35 +0100
Subject: [PATCH] debug

Change `test_Data` to `test_data` on line 174
---
 AttMIL_Training.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/AttMIL_Training.py b/AttMIL_Training.py
index 9d618ad..2c19fe5 100644
--- a/AttMIL_Training.py
+++ b/AttMIL_Training.py
@@ -171,7 +171,7 @@ def AttMIL_Training(args):
                     
                     model.load_state_dict(torch.load(os.path.join(args.projectFolder, 'RESULTS', 'bestModelFold' + str(foldcounter))))
                     model = model.to(device)
-                    test_dl = dls.test_dl(test_Data)
+                    test_dl = dls.test_dl(test_data)
                     probsList  = Validate_model_AttMIL(model = model, dataloaders = test_dl)
 
                     probs = {}
@@ -269,4 +269,4 @@ def AttMIL_Training(args):
         
         
         
-        
\ No newline at end of file
+