- input_size = 64, output_size = 8,
- depth = 5,
- cond = lambda d, k, s: k <= 4 and s <= k and (s == 1 or d < 3)
+ input_size=64,
+ output_size=8,
+ remain_depth=5,
+ # 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),