in mobile_cv/arch/fbnet_v2/irf_3d_block.py [0:0]
def __init__(
self,
in_channels,
out_channels,
*,
expansion=6,
kernel_size=3,
stride=1,
bias=False,
conv_args="conv3d",
bn_args="bn3d",
relu_args="relu",
se_args=None,
kernel_size_temporal=3,
stride_temporal=1,
res_conn_args="default",
# upsample_args="default",
width_divisor=8,
pw_args=None,
dw_args=None,
pwl_args=None,
skip_dw=False,
skip_pwl=False,
dw_skip_bnrelu=False,
pw_groups=1,
dw_group_ratio=1, # dw_group == mid_channels // dw_group_ratio
pwl_groups=1,
always_pw=False,
less_se_channels=False,
zero_last_bn_gamma=False,
drop_connect_rate=None,
mid_expand_out=False, # mid_channels = out_channels * expansion if mid_expand_out=True
last_relu=False, # apply relu after res_conn