mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
77131ea071
* save codes * save codes * support sar exporting to onnxruntime and tensorrt * use 4G max gpu space * support sequential sar * remove trt config for sar * add sar unit tests * add sar unit tests * resolve conflicts * remove torch.add rewrite in unit tests
14 lines
276 B
Python
14 lines
276 B
Python
_base_ = ['./text-recognition_static.py']
|
|
onnx_config = dict(
|
|
dynamic_axes={
|
|
'input': {
|
|
0: 'batch',
|
|
3: 'width'
|
|
},
|
|
'output': {
|
|
0: 'batch',
|
|
1: 'seq_len',
|
|
2: 'num_classes'
|
|
}
|
|
}, )
|