diff --git a/DBCNN.py b/DBCNN.py index b725959..b7e82c8 100644 --- 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() \ No newline at end of file + main()