From 519fbafcd948349cae2718a8d152a74a2030a203 Mon Sep 17 00:00:00 2001 From: co63oc Date: Fri, 16 Feb 2024 10:06:09 +0800 Subject: [PATCH] Update test_image_classification_fp16.py --- test/contrib/test_image_classification_fp16.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/contrib/test_image_classification_fp16.py b/test/contrib/test_image_classification_fp16.py index 63d157618a9e8..a4966feea8f3e 100644 --- a/test/contrib/test_image_classification_fp16.py +++ b/test/contrib/test_image_classification_fp16.py @@ -291,7 +291,7 @@ def infer(use_cuda, save_dirname=None): feed_target_names, fetch_targets, exe, - parogram=inference_program, + program=inference_program, clip_extra=True, )