Skip to content

Commit

Permalink
CI Failure addressed
Browse files Browse the repository at this point in the history
  • Loading branch information
siju-samuel committed Apr 5, 2020
1 parent 41495ac commit 03ada65
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions tests/python/frontend/tflite/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -1631,7 +1631,9 @@ def test_forward_mobilenet_v2():

def test_forward_mobilenet_v3():
"""Test the Mobilenet V3 TF Lite model."""
# MobilenetV3
# In MobilenetV3, some ops are not supported before tf 1.15 fbs schema
if package_version.parse(tf.VERSION) < package_version.parse('1.15.0'):
return
tflite_model_file = tf_testing.get_workload_official(
"https://storage.googleapis.com/mobilenet_v3/checkpoints/v3-large_224_1.0_float.tgz",
"v3-large_224_1.0_float/v3-large_224_1.0_float.tflite")
Expand Down Expand Up @@ -1747,7 +1749,10 @@ def test_forward_qnn_mobilenet_v2_net():

def test_forward_qnn_mobilenet_v3_net():
"""Test the Quantized TFLite Mobilenet V3 model."""
# MobilenetV3
# In MobilenetV3, some ops are not supported before tf 1.15 fbs schema
if package_version.parse(tf.VERSION) < package_version.parse('1.15.0'):
return

tflite_model_file = tf_testing.get_workload_official(
"https://storage.googleapis.com/mobilenet_v3/checkpoints/v3-large_224_1.0_uint8.tgz",
"v3-large_224_1.0_uint8/v3-large_224_1.0_uint8.tflite")
Expand Down

0 comments on commit 03ada65

Please sign in to comment.