Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[FRONTEND][Keras] Add test for MobileNet (apache#513)
Browse files Browse the repository at this point in the history
kazum authored and tqchen committed May 26, 2018
1 parent 029b0de commit aafe175
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions nnvm/tests/python/frontend/keras/test_forward.py
Original file line number Diff line number Diff line change
@@ -160,6 +160,12 @@ def test_forward_resnet50():
verify_keras_frontend(keras_model)


def test_forward_mobilenet():
keras_model = keras.applications.mobilenet.MobileNet(include_top=True, weights=None,
input_shape=(224,224,3), classes=1000)
verify_keras_frontend(keras_model)


if __name__ == '__main__':
test_forward_elemwise_add()
test_forward_softmax()
@@ -175,3 +181,4 @@ def test_forward_resnet50():
test_forward_vgg16()
test_forward_xception()
test_forward_resnet50()
test_forward_mobilenet()

0 comments on commit aafe175

Please sign in to comment.