typo bug in qnn test

apache · Apr 13, 2020 · 10bee0b · 10bee0b
1 parent 86c756e
commit 10bee0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/frontend/pytorch/qnn_test.py b/tests/python/frontend/pytorch/qnn_test.py
@@ -395,7 +395,7 @@ def get_imagenet_input():
         max_abs_diff = np.max(np.abs(tvm_result - pt_result))
         mean_abs_diff = np.mean(np.abs(tvm_result - pt_result))
         num_identical = np.sum(tvm_result == pt_result)
-        pt_top3_labels = np.argsort(pt_result)[::-1][:3]
+        pt_top3_labels = np.argsort(tvm_result)[::-1][:3]
         tvm_top3_labels = np.argsort(pt_result)[::-1][:3]
 
         print("\nModel name: %s" % model_name)