mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +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.
14 lines
306 B
YAML
14 lines
306 B
YAML
Models:
|
|
- Name: test_model
|
|
Metadata:
|
|
FLOPs: 319000000
|
|
Parameters: 3500000
|
|
Results:
|
|
- Dataset: ImageNet-1k
|
|
Metrics:
|
|
Top 1 Accuracy: 71.86
|
|
Top 5 Accuracy: 90.42
|
|
Task: Image Classification
|
|
Weights: test_weight.pth
|
|
Config: test_config.py
|