diff --git a/examples/post_training_quantization/openvino/yolov8_quantize_with_accuracy_control/main.py b/examples/post_training_quantization/openvino/yolov8_quantize_with_accuracy_control/main.py index 4b6cd2c0f75..4008ae27da7 100644 --- a/examples/post_training_quantization/openvino/yolov8_quantize_with_accuracy_control/main.py +++ b/examples/post_training_quantization/openvino/yolov8_quantize_with_accuracy_control/main.py @@ -38,7 +38,7 @@ def validate( ) -> Tuple[Dict, int, int]: validator.seen = 0 validator.jdict = [] - validator.stats = [] + validator.stats = dict(tp_m=[], tp=[], conf=[], pred_cls=[], target_cls=[]) validator.batch_i = 1 validator.confusion_matrix = ConfusionMatrix(nc=validator.nc) model.reshape({0: [1, 3, -1, -1]})