mmdeploy/configs/mmcls/classification_torchscript.py

7 lines
184 B
Python

_base_ = [
'../_base_/torchscript_config.py', '../_base_/backends/torchscript.py'
]
ir_config = dict(input_shape=None)
codebase_config = dict(type='mmcls', task='Classification')