guberti commented on code in PR #12539:
URL: https://github.com/apache/tvm/pull/12539#discussion_r951760286
##########
python/tvm/micro/testing/evaluation.py:
##########
@@ -154,6 +154,6 @@ def evaluate_model_accuracy(session, aot_executor,
input_data, true_labels, runs
aot_runtimes.append(runtime)
num_correct = sum(u == v for u, v in zip(true_labels, predicted_labels))
- average_time = sum(aot_runtimes) / len(aot_runtimes)
+ average_time = np.median(aot_runtimes)
Review Comment:
I think this is a great idea! To go one step further, I've expanded the
scope of this PR a little and reworked `evaluate_model_accuracy` into
`predict_labels_aot` - the description has been updated.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
