mmdeploy/configs/mmcls/mmcls_tensorrt.py

mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
_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)
add configs, add convert pipeline 2021-06-17 15:28:23 +08:00			`_base_ = ['./mmcls_base.py', '../_base_/backends/tensorrt.py']`
add tensorrt support (#2) 2021-06-23 13:14:28 +08:00			`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)`