Skip to content

Commit 94895ec

Browse files
authored
add build_neck and build_loss in model.__init__ (#237)
1 parent 20bf02b commit 94895ec

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

mmaction/models/__init__.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
from .backbones import (ResNet, ResNet2Plus1d, ResNet3d, ResNet3dCSN,
22
ResNet3dSlowFast, ResNet3dSlowOnly, ResNetTIN,
33
ResNetTSM)
4-
from .builder import (build_backbone, build_head, build_localizer, build_model,
5-
build_recognizer)
4+
from .builder import (build_backbone, build_head, build_localizer, build_loss,
5+
build_model, build_neck, build_recognizer)
66
from .common import Conv2plus1d
77
from .heads import BaseHead, I3DHead, SlowFastHead, TPNHead, TSMHead, TSNHead
88
from .localizers import BMN, PEM, TEM
@@ -20,5 +20,6 @@
2020
'ResNet3dSlowFast', 'SlowFastHead', 'Conv2plus1d', 'ResNet3dSlowOnly',
2121
'BCELossWithLogits', 'LOCALIZERS', 'build_localizer', 'PEM', 'TEM',
2222
'BinaryLogisticRegressionLoss', 'BMN', 'BMNLoss', 'build_model',
23-
'OHEMHingeLoss', 'SSNLoss', 'ResNet3dCSN', 'ResNetTIN', 'TPN', 'TPNHead'
23+
'OHEMHingeLoss', 'SSNLoss', 'ResNet3dCSN', 'ResNetTIN', 'TPN', 'TPNHead',
24+
'build_loss', 'build_neck'
2425
]

0 commit comments

Comments
 (0)