Skip to content

Commit b28c346

Browse files
author
Joey Tsai
committed
[Fix lint]
1 parent 7d92616 commit b28c346

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backends/qualcomm/quantizer/observers/per_channel_param_observer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from torch.ao.quantization.observer import UniformQuantizationObserverBase
33

44

5-
#TODO move to torch/ao/quantization/observer.py.
5+
# TODO move to torch/ao/quantization/observer.py.
66
class PerChannelParamObserver(UniformQuantizationObserverBase):
77
def __init__(
88
self,

examples/qualcomm/oss_scripts/fastvit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def main(args):
9191
quant_max=torch.iinfo(torch.int8).max,
9292
qscheme=torch.per_channel_symmetric,
9393
ch_axis=0,
94-
observer_or_fake_quant_ctr=ParamObserver.with_args(
94+
observer_or_fake_quant_ctr=PerChannelParamObserver.with_args(
9595
**{"steps": 200, "use_mse": True}
9696
),
9797
)

0 commit comments

Comments
 (0)