Skip to content

Commit

Permalink
Fixed output text in the ptq onnx example (#2272)
Browse files Browse the repository at this point in the history
### Changes

Fixed output text in the ptq onnx example

### Reason for changes

Incorrect output text in the ptq onnx example

### Related tickets

N/A

### Tests

test_examples
  • Loading branch information
alexsu52 authored Nov 15, 2023
1 parent 8efd04a commit 776670b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/post_training_quantization/onnx/mobilenet_v2/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,11 +140,11 @@ def transform_fn(data_item):
print("[4/7] Benchmark INT8 model:")
int8_fps = run_benchmark(int8_model_path, shape=[1, 3, 224, 224], verbose=True)

print("[5/7] Validate OpenVINO FP32 model:")
print("[5/7] Validate ONNX FP32 model in OpenVINO:")
fp32_top1 = validate(fp32_model_path, val_loader)
print(f"Accuracy @ top1: {fp32_top1:.3f}")

print("[6/7] Validate OpenVINO INT8 model:")
print("[6/7] Validate ONNX INT8 model in OpenVINO:")
int8_top1 = validate(int8_model_path, val_loader)
print(f"Accuracy @ top1: {int8_top1:.3f}")

Expand Down

0 comments on commit 776670b

Please sign in to comment.