mmdeploy/configs/mmaction/video-recognition/video-recognition_3d_tensorrt_static-256x256.py
Chen Xin 2c9861555f
[Enhancement] add mmaction.yml for test (#1455)
* add mmaction.yml for test

* t# This is a combination of 2 commits.

add missing file

* fix typo

* remove
2022-11-29 18:48:29 +08:00

15 lines
469 B
Python

_base_ = ['./video-recognition_static.py', '../../_base_/backends/tensorrt.py']
onnx_config = dict(input_shape=[256, 256])
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 30, 3, 32, 256, 256],
opt_shape=[1, 30, 3, 32, 256, 256],
max_shape=[1, 30, 3, 32, 256, 256])))
])