- if (n - 1) * stride + kernel_size == input_size:
- q = conv_chain(n, output_size, depth - 1, cond)
- r += [ [ (kernel_size, stride) ] + u for u in q ]
+ if (
+ n >= output_size
+ and (n - 1) * stride + kernel_size == input_size
+ ):
+ q = conv_chain(n, output_size, remain_depth - 1, cond)
+ r += [[(kernel_size, stride)] + u for u in q]