Commit Graph

4 Commits (e0e985fa85a6517e75845547b7e43f2908c3fa4d)

Author SHA1 Message Date
Jerry Jiarui XU faaf29e668
[Improvement] Move train_cfg/test_cfg inside model (#341)
* [Improvement] Move train_cfg/test_cfg inside model

* fixed config dict

* fixed config dict
2021-01-19 17:06:23 -08:00
Lei Yang f70507168f
Remove redundancies in pytorch2onnx (#160)
* rm redundancies

* re-add some packages
2020-09-30 10:11:34 +08:00
robin Han 0c04f52c42
Onnx upsample (#100)
* add customized Upsample which can convert to ONNX

* support multiply decode head for hrnet

* support size for Upsample
2020-09-03 19:59:13 +08:00
robin Han 9ba22456d9
add pytorch2onnx part (#12)
* add pytorch2onnx part

* Update according to the latest mmcv

* add docstring

* update docs

* update docs

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-08-14 03:28:21 +08:00