Commit a9d21497 by MORITA Kazutaka Committed by Tianqi Chen

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

parent c68945c5
......@@ -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()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment