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 6b7b738 commit 689c887Copy full SHA for 689c887
fms_mo/quant_refactor/base_quant.py
@@ -205,7 +205,7 @@ def __init__(
205
self.align_zero = align_zero
206
self.clipSTE = clipSTE
207
208
- temp_clipvals = torch.ones(self.perCh) if self.perCh else torch.Tensor([1.0])
+ temp_clipvals = torch.ones(self.qscheme.Nch) if self.perCh else torch.Tensor([1.0])
209
self.register_parameter("clip_val", torch.nn.Parameter(temp_clipvals.clone()))
210
# Keep clip_valn as positive 1.0 to allow simpler multiplication with
211
# negative numbers (clip_valn.data *= clip_valn)
0 commit comments