From 07f20ca1687f1ac60e0741991525fcf7d3a9b468 Mon Sep 17 00:00:00 2001
From: NBouteldja <40466985+NBouteldja@users.noreply.github.com>
Date: Wed, 23 Jun 2021 11:35:26 +0200
Subject: [PATCH] Update test.py

---
 test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.py b/test.py
index 460aefc..34ab746 100644
--- a/test.py
+++ b/test.py
@@ -45,7 +45,7 @@ logger = log.getLogger()
 logger.info('Script call arguments are:\n\n' + ' '.join(sys.argv[1:]) + '\n\n')
 
 segModel = Custom(input_ch=3, output_ch=8, modelDim=2)
-segModel.load_state_dict(torch.load('/homeStor1/nbouteldja/Results_ActivePAS/custom_train_val_test_e500_b6_r0.001_w1e-05_516_640_32_RAdam_instance_1deeper_Healthy_UUO_Adenine_Alport_IRI_NTN_fewSpecies_fewHuman_Background_+-1range_X/Model/finalModel.pt', map_location=lambda storage, loc: storage))
+segModel.load_state_dict(torch.load('<path-to-segmentation-model.pt>', map_location=lambda storage, loc: storage))
 segModel.eval()
 segModel = segModel.to(device)
 
-- 
GitLab