From: Francois Fleuret <francois@fleuret.org>
Date: Sun, 6 Jun 2021 13:14:40 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=4fc155881f00eda4365ff62e249138394f52a1cd

Update.
---

diff --git a/conv_chain.py b/conv_chain.py
index 85ae4fc..184e06b 100755
--- a/conv_chain.py
+++ b/conv_chain.py
@@ -31,7 +31,9 @@ if __name__ == "__main__":
     c = conv_chain(
         input_size = 64, output_size = 8,
         depth = 5,
-        cond = lambda d, k, s: k <= 4 and s <= k and (s == 1 or d < 3)
+        # We want kernels smaller than 4, strides smaller than the
+        # kernels, and stride of 1 except in the two last layers
+        cond = lambda d, k, s: k <= 4 and s <= k and (s == 1 or d <= 2)
     )
 
     x = torch.rand(1, 1, 64)