fix

intel-analytics · Sep 14, 2022 · b1dcaf0 · b1dcaf0
1 parent 1516d24
commit b1dcaf0
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/python/chronos/src/bigdl/chronos/forecaster/base_forecaster.py b/python/chronos/src/bigdl/chronos/forecaster/base_forecaster.py
@@ -1015,10 +1015,9 @@ def quantize(self, calib_data=None,
                quantization. You may choose from "mse", "mae", "rmse", "r2", "mape", "smape".
         :param conf: A path to conf yaml file for quantization. Default to None,
                using default config.
-        :param framework: string or list.
-               [{'pytorch_fx'|'pytorch_ipex'},
-                {'onnxrt_integerops'|'onnxrt_qlinearops'},
-                {'openvino'}] Default: 'pytorch_fx'. Consistent with Intel Neural Compressor.
+        :param framework: string or list. [{'pytorch_fx'|'pytorch_ipex'},
+               {'onnxrt_integerops'|'onnxrt_qlinearops'}, {'openvino'}]
+               Default: 'pytorch_fx'. Consistent with Intel Neural Compressor.
         :param approach: str, 'static' or 'dynamic'. Default to 'static'.
                OpenVINO supports static mode only, if set to 'dynamic',
                it will be replaced with 'static'.