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