File tree 2 files changed +3
-3
lines changed
examples/3.x_api/pytorch/cv/mixed_precision
neural_compressor/common/utils
2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -36,9 +36,9 @@ bash run_autotune.sh --input_model=resnet18 --dataset_location=/path/to/imagenet
36
36
## Benchmark
37
37
``` Shell
38
38
# run optimized performance
39
- bash run_benchmark.sh --input_model=resnet18 --dataset_location=/path/to/imagenet --mode=performance --batch_size=1 --optimized=true --iters=500
39
+ bash run_benchmark.sh --input_model=resnet18 --dataset_location=/path/to/imagenet --mode=performance --batch_size=100 --optimized=true --iters=500
40
40
# run optimized accuracy
41
- bash run_benchmark.sh --input_model=resnet18 --dataset_location=/path/to/imagenet --mode=accuracy --batch_size=100 --optimized=true
41
+ bash run_benchmark.sh --input_model=resnet18 --dataset_location=/path/to/imagenet --mode=accuracy --batch_size=1 --optimized=true
42
42
```
43
43
44
44
Original file line number Diff line number Diff line change 37
37
AUTOROUND = "autoround"
38
38
FP8_QUANT = "fp8_quant"
39
39
MX_QUANT = "mx_quant"
40
- MIX_PRECISION = "mix_precision "
40
+ MIXED_PRECISION = "mixed_precision "
41
41
42
42
# options
43
43
import datetime
You can’t perform that action at this time.
0 commit comments