def conv_filter_fn()

in utils/pipeline_utils_pixart.py [0:0]

• 4 lines of code
• 3 McCabe index (conditional complexity)

def conv_filter_fn(mod, *args):
    return (
        isinstance(mod, torch.nn.Conv2d) and mod.kernel_size == (1, 1) and 128 in [mod.in_channels, mod.out_channels]
    )