We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bea77c0 commit 30bb088Copy full SHA for 30bb088
1 file changed
nnvm/tests/python/frontend/keras/test_forward.py
@@ -160,6 +160,12 @@ def test_forward_resnet50():
160
verify_keras_frontend(keras_model)
161
162
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
169
if __name__ == '__main__':
170
test_forward_elemwise_add()
171
test_forward_softmax()
@@ -175,3 +181,4 @@ def test_forward_resnet50():
175
181
test_forward_vgg16()
176
182
test_forward_xception()
177
183
test_forward_resnet50()
184
+ test_forward_mobilenet()
0 commit comments