instance-seg_tvm-autotvm_static-800x1344.py 460 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
_base_ = [
    '../_base_/base_instance-seg_static.py', '../../_base_/backends/tvm.py'
]

onnx_config = dict(input_shape=[1344, 800])
backend_config = dict(model_inputs=[
    dict(
        use_vm=True,
        shape=dict(input=[1, 3, 800, 1344]),
        dtype=dict(input='float32'),
        tuner=dict(
            type='AutoTVMTuner',
            log_file='tvm_tune_log.log',
            n_trial=10000,
            tuner=dict(type='XGBTuner'),
        ))
])