mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
c127c474b9
* [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.
8 lines
242 B
Python
8 lines
242 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .inference import inference_model
|
|
from .model import ModelHub, get_model, init_model, list_models
|
|
|
|
__all__ = [
|
|
'init_model', 'inference_model', 'list_models', 'get_model', 'ModelHub'
|
|
]
|