mmdeploy/configs/mmcls/classification_ascend_dynam...

10 lines
261 B
Python

_base_ = ['./classification_dynamic.py', '../_base_/backends/ascend.py']
onnx_config = dict(input_shape=[224, 224])
backend_config = dict(model_inputs=[
dict(
dynamic_batch_size=[1, 2, 4, 8],
input_shapes=dict(input=[-1, 3, 224, 224]))
])