Update DBCNN.py

zwx8981 · Mar 7, 2019 · 500623e · 500623e
1 parent c5ebf79
commit 500623e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/DBCNN.py b/DBCNN.py
@@ -82,7 +82,7 @@ def forward(self, X):
         W2 = X2.size()[3]
         assert X2.size()[1] == 128        
 
-        if H != H2 | W != W2:
+        if (H != H2) | (W != W2):
             X2 = F.upsample_bilinear(X2,(H,W))
 
         X1 = X1.view(N, 512, H*W)
@@ -343,4 +343,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()