add int8 fp16 check

grimoire · Aug 31, 2020 · b61ed79 · b61ed79
1 parent eeba34f
commit b61ed79
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch2trt/torch2trt.py b/torch2trt/torch2trt.py
@@ -447,7 +447,8 @@ def torch2trt(module,
               int8_mode=False,
               int8_calib_dataset=None,
               int8_calib_algorithm=DEFAULT_CALIBRATION_ALGORITHM):
-
+    if int8_mode and fp16_mode:
+        fp16_mode = False
     inputs_in = inputs
 
     # copy inputs to avoid modifications to source data