mmpretrain/mmcls/models/classifiers
Ma Zerun c127c474b9
[Feature] Support getting model from the name defined in the model-index file. (#1236)
* [Feature] Support getting model from the name defined in the model-index file.

* Add unit tests.

* Prevent import `timm` if the `TIMMBackbone` is not used.

* Fix Windows CI.

* Move `init_model` to `mmcls.apis.hub`, and support pass nn.Module to all
model components.

* Fix requirements

* Rename `hub.py` to `model.py` and add unit tests.
2022-12-06 17:00:22 +08:00
..
__init__.py [Feature] Add TIMM and HuggingFace wrappers to build classifiers from them directly. (#1102) 2022-11-10 14:56:19 +08:00
base.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
hugging_face.py [Feature] Support getting model from the name defined in the model-index file. (#1236) 2022-12-06 17:00:22 +08:00
image.py [Feature] Support getting model from the name defined in the model-index file. (#1236) 2022-12-06 17:00:22 +08:00
timm.py [Feature] Support getting model from the name defined in the model-index file. (#1236) 2022-12-06 17:00:22 +08:00