Skip to content
Snippets Groups Projects
Commit bc0e193e authored by chenjun2hao's avatar chenjun2hao
Browse files

fix bugs

parent 79e13e56
No related branches found
No related tags found
No related merge requests found
......@@ -53,4 +53,4 @@ TEST:
FLIP_TEST: true
MULTI_SCALE: false
MODEL_FILE: "pretrained_models/best_val.pth"
OUTPUT_INDEX: 0
OUTPUT_INDEX: 1
......@@ -55,4 +55,4 @@ TEST:
SCALE_LIST: [1]
#0.5,0.75,1.0,1.25,1.5,1.75
# MODEL_FILE: "pretrained_models/best_val_smaller.pth"
OUTPUT_INDEX: 0
OUTPUT_INDEX: 1
......@@ -52,4 +52,4 @@ TEST:
FLIP_TEST: true
MULTI_SCALE: false
MODEL_FILE:
OUTPUT_INDEX: 0
OUTPUT_INDEX: 1
......@@ -337,7 +337,7 @@ class DualResNet(nn.Module):
if self.augment:
x_extra = self.seghead_extra(temp)
return [x_, x_extra]
return [x_extra, x_]
else:
return x_
......
......@@ -337,7 +337,7 @@ class DualResNet(nn.Module):
if self.augment:
x_extra = self.seghead_extra(temp)
return [x_, x_extra]
return [x_extra, x_]
else:
return x_
......
......@@ -357,7 +357,7 @@ class DualResNet(nn.Module):
if self.augment:
x_extra = self.seghead_extra(temp)
return [x_, x_extra]
return [x_extra, x_]
else:
return x_
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment