8cfc8ee03f
* add tensorrt temporal shift module and related pytorch implementations * add .gitignore and getn weights script. * rename get_wts.py script * Add tsm-r50 demo. * update readme * remove useless codes * update readme * update readme * remote video and .gitignore, update tutorial * update readme and tutorial * fix a few bugs and test on tensorrt 5.1 * update readme
22 lines
563 B
Python
22 lines
563 B
Python
# model settings
|
|
model = dict(
|
|
type='Recognizer2D',
|
|
backbone=dict(
|
|
type='ResNetTSM',
|
|
pretrained='torchvision://resnet50',
|
|
depth=50,
|
|
norm_eval=False,
|
|
shift_div=8),
|
|
cls_head=dict(
|
|
type='TSMHead',
|
|
num_classes=400,
|
|
in_channels=2048,
|
|
spatial_type='avg',
|
|
consensus=dict(type='AvgConsensus', dim=1),
|
|
dropout_ratio=0.5,
|
|
init_std=0.001,
|
|
is_shift=True),
|
|
# model training and testing settings
|
|
train_cfg=None,
|
|
test_cfg=dict(average_clips='prob'))
|