MCPcopy Create free account
hub / github.com/OpenBMB/VisRAG / __init__

Method __init__

timm_modified/timm/layers/selective_kernel.py:51–107  ·  view source on GitHub ↗

Selective Kernel Convolution Module As described in Selective Kernel Networks (https://arxiv.org/abs/1903.06586) with some modifications. Largest change is the input split, which divides the input channels across each convolution path, this can be viewed as a grouping of s

(self, in_channels, out_channels=None, kernel_size=None, stride=1, dilation=1, groups=1,
                 rd_ratio=1./16, rd_channels=None, rd_divisor=8, keep_3x3=True, split_input=True,
                 act_layer=nn.ReLU, norm_layer=nn.BatchNorm2d, aa_layer=None, drop_layer=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

__init__Method · 0.45

Calls 4

_kernel_validFunction · 0.85
ConvNormActAaClass · 0.85
make_divisibleFunction · 0.85
SelectiveKernelAttnClass · 0.85

Tested by

no test coverage detected