Mxbonn / INQ-pytorch

A PyTorch implementation of "Incremental Network Quantization: Towards Lossless CNNs with Low-Precision Weights"
164 stars 27 forks source link

can probability be set the same in random strategy? #14

Open hheavenknowss opened 3 years ago

hheavenknowss commented 3 years ago

Hi there, wonderful work. I have an question about random strategy, can I set the probability same in this model? like [0.2, 0.2, 0.6, 0.6, 0.8, 1]? I have read code in the quantization_scheduler.py, and in line 116: else: probability = (self.iterative_steps[self.idx] - self.iterative_steps[self.idx - 1]) / (1 - self.iterative_steps[self.idx - 1]) I think that means NO, but I have set the probability above, and the results not bad. I wonder what really happen here. can you help me plz?