mmclassification/hrnet-w40_4xb32_in1k.py at dd660ed99ee4eba797b31fc730ae95e032ef13e6 - mmclassification - Gitea: Git with a cup of tea

mirrors/mmclassification

mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 17:54:43 +08:00

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

7 lines

196 B

Python

Raw Blame History

 _base_ = [
     '../_base_/models/hrnet/hrnet-w40.py',
     '../_base_/datasets/imagenet_bs32_pil_resize.py',
     '../_base_/schedules/imagenet_bs256_coslr.py',
     '../_base_/default_runtime.py'
 ]