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