X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=ae_size.py;h=48dc2aff09cd8994f402ba3188fcf95937cf85c2;hp=71839378d3001d1c5b5eae1601d64b3509712995;hb=4f1a822b8d1b09bc2aed8c7fca99436db1f95620;hpb=0a30c677bfb0aac1f9eebc9646728519ca72bf5b

diff --git a/ae_size.py b/ae_size.py
index 7183937..48dc2af 100755
--- a/ae_size.py
+++ b/ae_size.py
@@ -25,11 +25,11 @@ if __name__ == "__main__":
     layer_specs = [ (11, 5), (5, 2), (3, 2), (3, 2) ]
 
     layers = []
-    for l in layer_specs:
-        layers.append(nn.Conv2d(1, 1, l[0], l[1]))
+    for kernel_size, stride in layer_specs:
+        layers.append(nn.Conv2d(1, 1, kernel_size, stride))
 
-    for l in reversed(layer_specs):
-        layers.append(nn.ConvTranspose2d(1, 1, l[0], l[1]))
+    for kernel_size, stride in reversed(layer_specs):
+        layers.append(nn.ConvTranspose2d(1, 1, kernel_size, stride))
 
     m = nn.Sequential(*layers)