Commit 9168476a authored by Bruno Korbar's avatar Bruno Korbar Committed by Francisco Massa
Browse files

[model cln] unused rMCX builder (#1188)

parent 9034749d
......@@ -36,35 +36,6 @@ def _mcX(model_depth, X=3, use_pool1=False, **kwargs):
return model
def _rmcX(model_depth, X=3, use_pool1=False, **kwargs):
"""Generate reverse mixed convolution network as in
https://arxiv.org/abs/1711.11248
Args:
model_depth (int): trunk depth - supports most resnet depths
X (int): Up to which layers are convolutions 2D
use_pool1 (bool, optional): Add pooling layer to the stem. Defaults to False.
Returns:
nn.Module: mcX video trunk
"""
assert X > 1 and X <= 5
conv_makers = [Conv3DNoTemporal] * (X - 2)
while len(conv_makers) < 5:
conv_makers.append(Conv3DSimple)
if model_depth < 50:
block = BasicBlock
else:
block = Bottleneck
model = VideoTrunkBuilder(block=block, conv_makers=conv_makers, model_depth=model_depth,
stem=get_default_stem(use_pool1=use_pool1), **kwargs)
return model
def mc3_18(use_pool1=False, **kwargs):
"""Constructor for 18 layer Mixed Convolution network as in
https://arxiv.org/abs/1711.11248
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment