segmentation_pplnn_static-512x1024.py 185 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
_base_ = ['./segmentation_static.py', '../_base_/backends/pplnn.py']

onnx_config = dict(input_shape=[1024, 512])

backend_config = dict(model_inputs=dict(opt_shape=[1, 3, 512, 1024]))