mmpretrain/tests/test_models
takuoko 9f31fc5390
[Refactor] Use `reset_classifier` to remove head of timm backbones. (#534)
* refactor timm backbone

* fix comment
2021-11-16 16:24:55 +08:00
..
test_backbones [Refactor] Use `reset_classifier` to remove head of timm backbones. (#534) 2021-11-16 16:24:55 +08:00
test_utils [Enhance] Enhence SE layer to support custom squeeze channels. (#417) 2021-08-20 13:31:44 +08:00
test_classifiers.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_heads.py [Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00
test_neck.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00