Skip to content

Commit 2aff100

Browse files
authored
Fixing imports in init file and preset name update (#2108)
* Fixing import in init file and preset name update * adapted the latest preset format style
1 parent 8ff601e commit 2aff100

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
from keras_hub.src.models.mobilenet.mobilenet_backbone import MobileNetBackbone
2+
from keras_hub.src.models.mobilenet.mobilenet_presets import backbone_presets
3+
from keras_hub.src.utils.preset_utils import register_presets
4+
5+
register_presets(backbone_presets, MobileNetBackbone)

keras_hub/src/models/mobilenet/mobilenet_presets.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
"""MobileNet preset configurations."""
22

33
backbone_presets = {
4-
"mobilenetv3_small_050": {
4+
"mobilenet_v3_small_050_imagenet": {
55
"metadata": {
66
"description": (
7-
"Small Mobilenet V3 model pre-trained on the ImageNet 1k "
7+
"Small MobileNet V3 model pre-trained on the ImageNet 1k "
88
"dataset at a 224x224 resolution."
99
),
1010
"params": 278784,

0 commit comments

Comments
 (0)