Skip to content
Snippets Groups Projects
Unverified Commit 07f20ca1 authored by NBouteldja's avatar NBouteldja Committed by GitHub
Browse files

Update test.py

parent 30dd396c
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
test.py
+ 1
− 1
View file @ 07f20ca1
...@@ -45,7 +45,7 @@ logger = log.getLogger() ...@@ -45,7 +45,7 @@ logger = log.getLogger()
logger.info('Script call arguments are:\n\n' + ' '.join(sys.argv[1:]) + '\n\n') 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 = 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.eval()
segModel = segModel.to(device) segModel = segModel.to(device)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment