Function _kernel_valid

lib/models_timm/layers/selective_kernel.py:15–19

(k)

def _kernel_valid(k):
    if isinstance(k, (list, tuple)):
        for ki in k:
            return _kernel_valid(ki)
    assert k >= 3 and k % 2


class SelectiveKernelAttn(nn.Module):