diff --git a/tools/validate.py b/tools/validate.py index d46284dcfc01067cbd2641877592c107cce8f460..9cf27e31cd91e21422b6e47d0723e9c8f1a9439b 100644 --- a/tools/validate.py +++ b/tools/validate.py @@ -37,6 +37,7 @@ def compare_output(output_name, mace_out_value, out_value): similarity = (1 - spatial.distance.cosine(out_value, mace_out_value)) print output_name, 'MACE VS', FLAGS.platform.upper(), 'similarity: ', similarity if (FLAGS.mace_runtime == "cpu" and similarity > 0.999) or \ + (FLAGS.mace_runtime == "neon" and similarity > 0.999) or \ (FLAGS.mace_runtime == "gpu" and similarity > 0.995) or \ (FLAGS.mace_runtime == "dsp" and similarity > 0.930): print '=======================Similarity Test Passed======================'