# We want kernels smaller than 4, strides smaller than the
# kernels, and strides of 1 except in the two last layers
cond = lambda d, k, s: k <= 4 and s <= k and (s == 1 or d <= 2)
# We want kernels smaller than 4, strides smaller than the
# kernels, and strides of 1 except in the two last layers
cond = lambda d, k, s: k <= 4 and s <= k and (s == 1 or d <= 2)