Skip to content
Snippets Groups Projects
Commit a9d21497 authored by MORITA Kazutaka's avatar MORITA Kazutaka Committed by Tianqi Chen
Browse files

[FRONTEND][Keras] Add test for MobileNet (#513)

parent c68945c5
No related branches found
No related tags found
No related merge requests found
......@@ -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 @@ if __name__ == '__main__':
test_forward_vgg16()
test_forward_xception()
test_forward_resnet50()
test_forward_mobilenet()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment