File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -51,7 +51,7 @@ class FakeQuantizeConfig:
51
51
zero_point_precision: zero point dtype (default torch.int32)
52
52
zero_point_domain: whether zero point is in integer (default) or float domain
53
53
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
55
55
(default false), not compatible with `is_dynamic`.
56
56
57
57
kwargs (optional):
@@ -406,7 +406,7 @@ def initialize_fake_quantizers(
406
406
example_inputs : Tuple [Any , ...],
407
407
) -> None :
408
408
"""
409
- Initialize the scales and zero points on all
409
+ (Prototype) Initialize the scales and zero points on all
410
410
:class:`~`torchao.quantization.qat.fake_quantizer.FakeQuantizer`
411
411
in the model based on the provided example inputs.
412
412
"""
You can’t perform that action at this time.
0 commit comments