Commit Graph

4 Commits (main)

Author SHA1 Message Date
Haian Huang(深度眸) 927e0a46af
Support setting the cache_size_limit parameter and support mmdet 3.0.0 (#707)
* Support setting the cache_size_limit parameter of dynamo in PyTorch 2.0

* update

* update

* fix

* fix

* update

* update

* update version
2023-04-18 10:59:52 +08:00
Nioolek e966ce4e60
[Improve] Beauty RTMDet config (#531)
* beauty rtmdet config

* format
2023-02-09 19:08:21 +08:00
Haian Huang(深度眸) 884330108d
[BC Breaking] Keep the same input shape order as mmdet 3.0.0rc5 (#436)
* Keep the same input shape order as mmdet

* add desc

* update

* update
2023-01-06 19:00:30 +08:00
RangiLyu 48f8896e84
[Feature] Implement fast version of RTMDet. (#425)
* Accelerate RTMDet

* update

* update

* update

* update1

* update2

* update pipeline

* update lr cudnnbenchmark

* revert batchsize

* fix batch inference

* refactor head

* update box

* bs=16

* update

* move reduce mean

* update head

* per img loss

* fix

* fix sum

* concat loss

* batch dsla

* sort topk

* bs 32

* clean code

* update readme

* update ut

* update checkpoint

* num_class

* clean code

* resolve comments

* fix readme

* fix ut

Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-04 16:06:44 +08:00