Skip to content

Commit dacd3aa

Browse files
authored
Mark QAT range learning as prototype for now (#2272)
API may be subject to change. See this tracker for future tasks for more detail: #2271
1 parent 1b61c82 commit dacd3aa

File tree

1 file changed

+2
-2
lines changed
  • torchao/quantization/qat

1 file changed

+2
-2
lines changed

torchao/quantization/qat/api.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ class FakeQuantizeConfig:
5151
zero_point_precision: zero point dtype (default torch.int32)
5252
zero_point_domain: whether zero point is in integer (default) or float domain
5353
is_dynamic: whether to use dynamic (default) or static scale and zero points
54-
range_learning: whether to learn scale and zero points during training
54+
range_learning (prototype): whether to learn scale and zero points during training
5555
(default false), not compatible with `is_dynamic`.
5656
5757
kwargs (optional):
@@ -406,7 +406,7 @@ def initialize_fake_quantizers(
406406
example_inputs: Tuple[Any, ...],
407407
) -> None:
408408
"""
409-
Initialize the scales and zero points on all
409+
(Prototype) Initialize the scales and zero points on all
410410
:class:`~`torchao.quantization.qat.fake_quantizer.FakeQuantizer`
411411
in the model based on the provided example inputs.
412412
"""

0 commit comments

Comments
 (0)