Skip to content

Commit 30bb088

Browse files
kazumsergei-mironov
authored andcommitted
[FRONTEND][Keras] Add test for MobileNet (apache#513)
1 parent bea77c0 commit 30bb088

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

nnvm/tests/python/frontend/keras/test_forward.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,12 @@ def test_forward_resnet50():
160160
verify_keras_frontend(keras_model)
161161

162162

163+
def test_forward_mobilenet():
164+
keras_model = keras.applications.mobilenet.MobileNet(include_top=True, weights=None,
165+
input_shape=(224,224,3), classes=1000)
166+
verify_keras_frontend(keras_model)
167+
168+
163169
if __name__ == '__main__':
164170
test_forward_elemwise_add()
165171
test_forward_softmax()
@@ -175,3 +181,4 @@ def test_forward_resnet50():
175181
test_forward_vgg16()
176182
test_forward_xception()
177183
test_forward_resnet50()
184+
test_forward_mobilenet()

0 commit comments

Comments
 (0)