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, )