mmdeploy/configs/mmseg/segmentation_tensorrt-int8_dynamic-512x1024-2048x2048.py
AllentDan bca8cc0897
[fix] align the names of configs (#256)
* align names of configs

* set input_shape to None for all ppl
2021-12-07 21:06:44 +08:00

12 lines
403 B
Python

_base_ = ['./segmentation_dynamic.py', '../_base_/backends/tensorrt-int8.py']
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 512, 1024],
opt_shape=[1, 3, 1024, 2048],
max_shape=[1, 3, 2048, 2048])))
])