Merge branch 'master' of https://github.com/pedrodiamel/pytorchvision

pedrodiamel · Jun 22, 2019 · c58f632 · c58f632
2 parents 0fcc4bb + 9725438
commit c58f632
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytvision/netmodels/dexpression.py b/pytvision/netmodels/dexpression.py
@@ -36,7 +36,7 @@ def __init__(self, num_classes=8, num_channels=1 ):
 
         self.num_classes = num_classes
         self.num_channels = num_channels
-        self.size_input=48 
+        self.size_input=224 
         self.dim = 272*14*14
 
         self.conv01 = BasicConv2d(num_channels, 64, kernel_size=7, stride=2, padding=3)