X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=ae_size.py;h=8afb101100b87429351aa4dc157f59973ec398c1;hp=25ecc4a13222f176356028c93e32a69858e413d9;hb=e111e52ea09f03fc309052b00a452ced52668f70;hpb=bbc4ab46de8cb2e05f83e35e15f3adfd23490089

diff --git a/ae_size.py b/ae_size.py
index 25ecc4a..8afb101 100755
--- a/ae_size.py
+++ b/ae_size.py
@@ -12,9 +12,9 @@ def minimal_input_size(w, layer_specs):
         return w
     else:
         kernel_size, stride = layer_specs[0]
-        w = math.ceil((w - kernel_size) / stride) + 1
-        w = minimal_input_size(w, layer_specs[1:])
-        return int((w - 1) * stride + kernel_size)
+        v = int(math.ceil((w - kernel_size) / stride)) + 1
+        v = minimal_input_size(v, layer_specs[1:])
+        return (v - 1) * stride + kernel_size
 
 ######################################################################
 
@@ -22,9 +22,10 @@ def minimal_input_size(w, layer_specs):
 
 if __name__ == "__main__":
 
-    layer_specs = [ (11, 5), (5, 4), (3, 2), (3, 2) ]
+    layer_specs = [ (17, 5), (5, 4), (3, 2), (3, 2) ]
 
     layers = []
+
     for kernel_size, stride in layer_specs:
         layers.append(nn.Conv2d(1, 1, kernel_size, stride))