"vscode:/vscode.git/clone" did not exist on "9886c7001596437a5ff0de646f74b334f0932504"
classification_coreml_dynamic-224x224-224x224.py 383 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
8
9
10
11
12
_base_ = ['../_base_/torchscript_config.py', '../_base_/backends/coreml.py']

codebase_config = dict(type='mmpretrain', task='Classification')

backend_config = dict(model_inputs=[
    dict(
        input_shapes=dict(
            input=dict(
                min_shape=[1, 3, 224, 224],
                max_shape=[8, 3, 224, 224],
                default_shape=[1, 3, 224, 224])))
])