fix ceil mode bug

tuvia0213 · Jan 23, 2019 · 6d01fd4 · 6d01fd4
1 parent 486965c
commit 6d01fd4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dface/core/models.py b/dface/core/models.py
@@ -121,7 +121,7 @@ def __init__(self,is_train=False, use_cuda=True):
             nn.PReLU()  # prelu3
 
         )
-        self.conv4 = nn.Linear(64*2*2, 128)  # conv4
+        self.conv4 = nn.Linear(64*3*3, 128)  # conv4
         self.prelu4 = nn.PReLU()  # prelu4
         # detection
         self.conv5_1 = nn.Linear(128, 1)
@@ -172,7 +172,7 @@ def __init__(self,is_train=False, use_cuda=True):
             nn.Conv2d(64,128,kernel_size=2,stride=1), # conv4
             nn.PReLU() # prelu4
         )
-        self.conv5 = nn.Linear(128*2*2, 256)  # conv5
+        self.conv5 = nn.Linear(128*3*3, 256)  # conv5
         self.prelu5 = nn.PReLU()  # prelu5
         # detection
         self.conv6_1 = nn.Linear(256, 1)