Skip to content

Commit 9168476

Browse files
bjuncekfmassa
authored andcommitted
[model cln] unused rMCX builder (#1188)
1 parent 9034749 commit 9168476

File tree

1 file changed

+0
-29
lines changed

1 file changed

+0
-29
lines changed

torchvision/models/video/mixed_conv.py

Lines changed: 0 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -36,35 +36,6 @@ def _mcX(model_depth, X=3, use_pool1=False, **kwargs):
3636
return model
3737

3838

39-
def _rmcX(model_depth, X=3, use_pool1=False, **kwargs):
40-
"""Generate reverse mixed convolution network as in
41-
https://arxiv.org/abs/1711.11248
42-
43-
Args:
44-
model_depth (int): trunk depth - supports most resnet depths
45-
X (int): Up to which layers are convolutions 2D
46-
use_pool1 (bool, optional): Add pooling layer to the stem. Defaults to False.
47-
48-
Returns:
49-
nn.Module: mcX video trunk
50-
"""
51-
assert X > 1 and X <= 5
52-
53-
conv_makers = [Conv3DNoTemporal] * (X - 2)
54-
while len(conv_makers) < 5:
55-
conv_makers.append(Conv3DSimple)
56-
57-
if model_depth < 50:
58-
block = BasicBlock
59-
else:
60-
block = Bottleneck
61-
62-
model = VideoTrunkBuilder(block=block, conv_makers=conv_makers, model_depth=model_depth,
63-
stem=get_default_stem(use_pool1=use_pool1), **kwargs)
64-
65-
return model
66-
67-
6839
def mc3_18(use_pool1=False, **kwargs):
6940
"""Constructor for 18 layer Mixed Convolution network as in
7041
https://arxiv.org/abs/1711.11248

0 commit comments

Comments
 (0)