mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
6 lines
215 B
Python
6 lines
215 B
Python
_base_ = ['./mmcls_base.py', '../_base_/backends/tensorrt.py']
|
|
tensorrt_param = dict(
|
|
opt_shape_dict=dict(
|
|
input=[[1, 3, 224, 224], [4, 3, 224, 224], [32, 3, 224, 224]]),
|
|
max_workspace_size=1 << 30)
|