Skip to content

GitLab

Unverified Commit 93dbaf3e authored by lin bin's avatar lin bin Committed by GitHub
Browse files

fix precision type (#3893)

parent a4760ce8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
nni/compression/pytorch/quantization_speedup/integrated_tensorrt.py
View file @ 93dbaf3e
...@@ -24,7 +24,7 @@ class CalibrateType: ...@@ -24,7 +24,7 @@ class CalibrateType:
MINMAX = trt.CalibrationAlgoType.MINMAX_CALIBRATION MINMAX = trt.CalibrationAlgoType.MINMAX_CALIBRATION
Precision_Dict = { Precision_Dict = {
8: trt.float32, 8: trt.int8,
16: trt.float16, 16: trt.float16,
32: trt.float32 32: trt.float32
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment