q.yao 6ccb1c0fe5 [Feature] add onnxruntime test tool (#498) ...

* add onnxruntime test tool, update pytorch2onnx to support slice export

* onnx convert with custom output shape, update test code

* update pytorch2onnx, add rescale_shape support, add document

* update doc for lint error fixing

* remove cpu flag in ort_test.py

* change class name, fix cuda error

* remote comment

* fix bug of torch2onnx

* mIOU to mIoU

2021-04-28 20:38:01 -07:00

..

convert_datasets

Doc refactor (#311)

2020-12-22 18:36:49 -08:00

analyze_logs.py

add plot_logs tool (#426)

2021-03-21 22:05:32 -07:00

benchmark.py

[Bug fix] fixed fp16 inference (#497)

2021-04-24 09:57:33 -07:00

dist_test.sh

Add test tutorial (#9)

2020-07-12 23:53:56 +08:00

dist_train.sh

init commit

2020-07-10 02:39:01 +08:00

get_flops.py

[Improvement] Move train_cfg/test_cfg inside model (#341)

2021-01-19 17:06:23 -08:00

ort_test.py

[Feature] add onnxruntime test tool (#498)

2021-04-28 20:38:01 -07:00

print_config.py

add print model graph args for tools/print_config.py (#451)

2021-04-06 15:14:54 -07:00

publish_model.py

init commit

2020-07-10 02:39:01 +08:00

pytorch2onnx.py

[Feature] add onnxruntime test tool (#498)

2021-04-28 20:38:01 -07:00

pytorch2torchscript.py

add tool pytorch2torchscript (#469)

2021-04-19 08:51:49 -07:00

slurm_test.sh

init commit

2020-07-10 02:39:01 +08:00

slurm_train.sh

init commit

2020-07-10 02:39:01 +08:00

test.py

[Bug fix] fixed fp16 inference (#497)

2021-04-24 09:57:33 -07:00

train.py

[Improvement] Move train_cfg/test_cfg inside model (#341)

2021-01-19 17:06:23 -08:00