Thanks!
If the config of the conv layers differs in some attributes, e.g. kernel_size, how would I need to use it, as it seems I cannot overwrite k in the stacked calls like:
flatten(*flatten(w=100, k=3, s=1, p=0, m=True), k=5)

Also, the pooling operation seems to be fixed to kernel_size==stride=2 and the pooling argument also propagates?