Skip to content

Commit

Permalink
Set random seed for test
Browse files Browse the repository at this point in the history
Change-Id: I542a787d11422ea83c52147b2cb1144fcef0dd77
  • Loading branch information
mbaret committed Jan 6, 2020
1 parent ffb3234 commit 3dcba97
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/python/frontend/tflite/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -1140,6 +1140,7 @@ def test_detection_postprocess():
converter.allow_custom_ops = True
converter.inference_type = tf.lite.constants.FLOAT
tflite_model = converter.convert()
np.random.seed(0)
box_encodings = np.random.uniform(size=(1, 1917, 4)).astype('float32')
class_predictions = np.random.uniform(size=(1, 1917, 91)).astype('float32')
tflite_output = run_tflite_graph(tflite_model, [box_encodings, class_predictions])
Expand Down

0 comments on commit 3dcba97

Please sign in to comment.