ZouJiu1 / LSQplus

LSQ+ or LSQplus
57 stars 14 forks source link

batch_init初始化代码有误 #16

Open MJITG opened 7 months ago

MJITG commented 7 months ago

https://github.com/ZouJiu1/LSQplus/blob/b8c6ea041d1678097c97cc77027e06c26d898d06/quantization/lsqplus_quantize_V2.py#L140C38-L140C38

朋友您好,这行代码应该是:

self.beta.data = self.beta.data*0.9 + 0.1* (mina - self.s.data * self.Qn)