Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Chronos: fix quantize and to_torch_data_loader API doc #5755

Merged
merged 2 commits into from
Sep 15, 2022
Merged

Chronos: fix quantize and to_torch_data_loader API doc #5755

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
48853dd
fix doc
plusbang Sep 14, 2022
3f431dd
fix
plusbang Sep 14, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions python/chronos/src/bigdl/chronos/data/tsdataset.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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.

Expand Down
7 changes: 3 additions & 4 deletions python/chronos/src/bigdl/chronos/forecaster/base_forecaster.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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'|'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'.
Expand Down