From 3172ad88a85e655f5d5ea2b07baf451c9e9da2bc Mon Sep 17 00:00:00 2001 From: xqqin Date: Wed, 13 Sep 2023 04:20:32 +0000 Subject: [PATCH] add print for Max probability --- demo/test_thinker/test_thinker.c | 1 + 1 file changed, 1 insertion(+) diff --git a/demo/test_thinker/test_thinker.c b/demo/test_thinker/test_thinker.c index 808bfc4..ff372e6 100644 --- a/demo/test_thinker/test_thinker.c +++ b/demo/test_thinker/test_thinker.c @@ -180,6 +180,7 @@ int thinker_task_test(int loop_count, char *argv[]) } } + printf("Max probability: %d\n", max_probability); printf("Predicted category index: %d\n", predicted_category_index); printf("Predicted label: %s\n", classes[predicted_category_index]);