diff --git a/examples/3.x_api/pytorch/cv/mixed_precision/README.md b/examples/3.x_api/pytorch/cv/mixed_precision/README.md index 3281801578f..597c9e2a3fb 100644 --- a/examples/3.x_api/pytorch/cv/mixed_precision/README.md +++ b/examples/3.x_api/pytorch/cv/mixed_precision/README.md @@ -36,9 +36,9 @@ bash run_autotune.sh --input_model=resnet18 --dataset_location=/path/to/imagenet ## Benchmark ```Shell # run optimized performance -bash run_benchmark.sh --input_model=resnet18 --dataset_location=/path/to/imagenet --mode=performance --batch_size=1 --optimized=true --iters=500 +bash run_benchmark.sh --input_model=resnet18 --dataset_location=/path/to/imagenet --mode=performance --batch_size=100 --optimized=true --iters=500 # run optimized accuracy -bash run_benchmark.sh --input_model=resnet18 --dataset_location=/path/to/imagenet --mode=accuracy --batch_size=100 --optimized=true +bash run_benchmark.sh --input_model=resnet18 --dataset_location=/path/to/imagenet --mode=accuracy --batch_size=1 --optimized=true ``` diff --git a/neural_compressor/common/utils/constants.py b/neural_compressor/common/utils/constants.py index 629a3f5743e..86a4c342f58 100644 --- a/neural_compressor/common/utils/constants.py +++ b/neural_compressor/common/utils/constants.py @@ -37,7 +37,7 @@ AUTOROUND = "autoround" FP8_QUANT = "fp8_quant" MX_QUANT = "mx_quant" -MIX_PRECISION = "mix_precision" +MIXED_PRECISION = "mixed_precision" # options import datetime