Skip to content

Commit c89a186

Browse files
usharengushareng
authored andcommitted
mobilenet_v3 added in keras-nlp (keras-team#1782)
* mobilenet_v3 added in keras-nlp * minor bug fixed in mobilenet_v3_backbone * formatting corrected * refactoring backbone * correct_pad_downsample method added * refactoring backbone * parameters updated * Testcaseupdated, expected output shape corrected * code formatted with black * testcase updated * refactoring and description added * comments updated * added mobilenet v1 and v2 * merge conflict resolved * version arg removed, and config options added * input_shape changed to image_shape in arg * config updated * input shape corrected * comments resolved * activation function format changed * minor bug fixed * minor bug fixed * added vision_backbone_test * channel_first bug resolved * channel_first cases working * comments resolved * formatting fixed * refactoring --------- Co-authored-by: ushareng <[email protected]>
1 parent 0ecf860 commit c89a186

File tree

6 files changed

+790
-0
lines changed

6 files changed

+790
-0
lines changed

keras_nlp/api/models/__init__.py

+4
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,10 @@
195195
from keras_nlp.src.models.mix_transformer.mix_transformer_classifier import (
196196
MiTImageClassifier,
197197
)
198+
from keras_nlp.src.models.mobilenet.mobilenet_backbone import MobileNetBackbone
199+
from keras_nlp.src.models.mobilenet.mobilenet_image_classifier import (
200+
MobileNetImageClassifier,
201+
)
198202
from keras_nlp.src.models.opt.opt_backbone import OPTBackbone
199203
from keras_nlp.src.models.opt.opt_causal_lm import OPTCausalLM
200204
from keras_nlp.src.models.opt.opt_causal_lm_preprocessor import (
+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
# Copyright 2024 The KerasNLP Authors
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# https://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.

0 commit comments

Comments
 (0)