From 2fb344954ca30eb44fa223372f288c93b03ae2ab Mon Sep 17 00:00:00 2001 From: yejianwu Date: Fri, 30 Mar 2018 17:51:19 +0800 Subject: [PATCH] add neon validate similarity --- tools/validate.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/validate.py b/tools/validate.py index d46284dc..9cf27e31 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======================' -- GitLab