We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7d92616 commit b28c346Copy full SHA for b28c346
backends/qualcomm/quantizer/observers/per_channel_param_observer.py
@@ -2,7 +2,7 @@
2
from torch.ao.quantization.observer import UniformQuantizationObserverBase
3
4
5
-#TODO move to torch/ao/quantization/observer.py.
+# TODO move to torch/ao/quantization/observer.py.
6
class PerChannelParamObserver(UniformQuantizationObserverBase):
7
def __init__(
8
self,
examples/qualcomm/oss_scripts/fastvit.py
@@ -91,7 +91,7 @@ def main(args):
91
quant_max=torch.iinfo(torch.int8).max,
92
qscheme=torch.per_channel_symmetric,
93
ch_axis=0,
94
- observer_or_fake_quant_ctr=ParamObserver.with_args(
+ observer_or_fake_quant_ctr=PerChannelParamObserver.with_args(
95
**{"steps": 200, "use_mse": True}
96
),
97
)
0 commit comments