From 4fc155881f00eda4365ff62e249138394f52a1cd Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Sun, 6 Jun 2021 15:14:40 +0200 Subject: [PATCH] Update. --- conv_chain.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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) -- 2.39.5