Skip to content

Commit 5fe9c62

Browse files
Add TinyNet and list_models (#5)
* Fix efficientv2 b3 * Add `list_models`
1 parent f55d5ac commit 5fe9c62

14 files changed

+495
-30
lines changed

kimm/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,4 @@
1+
from kimm import models # force to add models to the registry
2+
from kimm.utils.model_registry import list_models
3+
14
__version__ = "0.1.0"

kimm/models/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,7 @@
1+
from kimm.models.efficientnet import * # noqa:F403
12
from kimm.models.feature_extractor import FeatureExtractor
23
from kimm.models.ghostnet import * # noqa:F403
4+
from kimm.models.mobilenet_v2 import * # noqa:F403
5+
from kimm.models.mobilenet_v3 import * # noqa:F403
36
from kimm.models.resnet import * # noqa:F403
7+
from kimm.models.vision_transformer import * # noqa:F403

0 commit comments

Comments
 (0)