q.yao 805eda81ea

add onnx to tensorrt tools (#542)

2021-05-11 20:02:27 -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

onnx2tensorrt.py

add onnx to tensorrt tools (#542)

2021-05-11 20:02:27 -07: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