From 48853dde3d63c1f3fb09f5639260c026ec2b9cbc Mon Sep 17 00:00:00 2001 From: binbin Date: Wed, 14 Sep 2022 14:40:35 +0800 Subject: [PATCH 1/2] fix doc --- python/chronos/src/bigdl/chronos/data/tsdataset.py | 6 +++--- .../chronos/src/bigdl/chronos/forecaster/base_forecaster.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/python/chronos/src/bigdl/chronos/data/tsdataset.py b/python/chronos/src/bigdl/chronos/data/tsdataset.py index 9910632f6e9..597eaaa5b9c 100644 --- a/python/chronos/src/bigdl/chronos/data/tsdataset.py +++ b/python/chronos/src/bigdl/chronos/data/tsdataset.py @@ -722,9 +722,9 @@ def to_torch_data_loader(self, This parameter should be set to True only when you are using Autoformer model. This indicates the length of overlap area of output(y) and input(x) on time axis. :param is_predict: bool, - This parameter should be set to True only when you are using Autoformer model. This - indicates if the dataset will be sampled as a prediction dataset(without groud - truth). + This parameter should be set to True only when you are processing test data without + accuracy evaluation for Autoformer model. This indicates if the dataset will be + sampled as a prediction dataset(without groud truth). :return: A pytorch DataLoader instance. diff --git a/python/chronos/src/bigdl/chronos/forecaster/base_forecaster.py b/python/chronos/src/bigdl/chronos/forecaster/base_forecaster.py index 868affc74ff..66a8ef45a58 100644 --- a/python/chronos/src/bigdl/chronos/forecaster/base_forecaster.py +++ b/python/chronos/src/bigdl/chronos/forecaster/base_forecaster.py @@ -1016,7 +1016,7 @@ def quantize(self, calib_data=None, :param conf: A path to conf yaml file for quantization. Default to None, using default config. :param framework: string or list. - [{'pytorch'|'pytorch_fx'|'pytorch_ipex'}, + [{'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'. From 3f431ddead762d2a4116e88fdec8aa2936a32534 Mon Sep 17 00:00:00 2001 From: binbin Date: Wed, 14 Sep 2022 15:49:49 +0800 Subject: [PATCH 2/2] fix --- .../src/bigdl/chronos/forecaster/base_forecaster.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/python/chronos/src/bigdl/chronos/forecaster/base_forecaster.py b/python/chronos/src/bigdl/chronos/forecaster/base_forecaster.py index 66a8ef45a58..89ed62f77d1 100644 --- 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'.