mmpretrain/mmcls/models
Ma Zerun 5de480ea9e
[Feature] Support HRNet and add pre-trained models (#660)
* Support HRNet

* Add HRNet configs

* Fix a bug in backward

* Add configs and update docs.

* Not use bias in conv before batch norm

* Defaults to use `norm_eval=False`

* Add unit tests and support out_channels in HRFuseScales

* Update checkpoint path

* Update docstring.

* Remove incorrect files

* Improve according to comments
2022-01-28 10:54:14 +08:00
..
backbones [Feature] Support HRNet and add pre-trained models (#660) 2022-01-28 10:54:14 +08:00
classifiers [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
heads [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
losses [Enhance] Support single-label, softmax, custom eps by asymmetric loss (#609) 2022-01-21 11:35:14 +08:00
necks [Feature] Support HRNet and add pre-trained models (#660) 2022-01-28 10:54:14 +08:00
utils [Feature] Add Twins backbone and convert checkpoints. (#642) 2022-01-27 10:25:05 +08:00
__init__.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
builder.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00