initial commit

Reagan1311 · Jul 29, 2019 · e50e103 · e50e103
1 parent 334b97f
commit e50e103
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/model/DABNet.py b/model/DABNet.py
@@ -161,14 +161,12 @@ def forward(self, input):
 
         # DAB Block 1
         output1_0 = self.downsample_1(output0_cat)
-        output1 = output1_0
-        output1 = self.DAB_Block_1(output1)
+        output1 = self.DAB_Block_1(output1_0)
         output1_cat = self.bn_prelu_2(torch.cat([output1, output1_0, down_2], 1))
 
         # DAB Block 2
         output2_0 = self.downsample_2(output1_cat)
-        output2 = output2_0
-        output2 = self.DAB_Block_2(output2)
+        output2 = self.DAB_Block_2(output2_0)
         output2_cat = self.bn_prelu_3(torch.cat([output2, output2_0, down_3], 1))
 
         out = self.classifier(output2_cat)

diff --git a/test.py b/test.py
@@ -83,9 +83,9 @@ def test_model(args):
                 checkpoint = torch.load(args.checkpoint)
                 model.load_state_dict(checkpoint['model'])
                 # model.load_state_dict(convert_state_dict(checkpoint['model']))
-        else:
-            print("=====> no checkpoint found at '{}'".format(args.checkpoint))
-            raise FileNotFoundError("no checkpoint found at '{}'".format(args.checkpoint))
+            else:
+                print("=====> no checkpoint found at '{}'".format(args.checkpoint))
+                raise FileNotFoundError("no checkpoint found at '{}'".format(args.checkpoint))
 
         print("=====> beginning validation")
         print("validation set length: ", len(testLoader))