Jerry Jiarui XU 0c5b026db1

[Refactor]: Unified parameter initialization (#567) ...

* [Refactor]: Unified parameter initialization

* fixed pretrained

2021-06-16 21:41:29 -07:00

..

convert_datasets

…

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

deploy_test.py

[Feature] Update deploy test tools (#553)

2021-05-24 20:37:46 -07:00

dist_test.sh

…

dist_train.sh

…

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

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

…

pytorch2onnx.py

fix verify bugs (#547)

2021-05-12 21:14:41 -07:00

pytorch2torchscript.py

add tool pytorch2torchscript (#469)

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

slurm_test.sh

…

slurm_train.sh

…

test.py

[feature]: Able to use save_best option (#575)

2021-06-02 16:25:26 -07:00

train.py

[Refactor]: Unified parameter initialization (#567)

2021-06-16 21:41:29 -07:00