Bug in UniformQuantize class #4
Description
Hi, thank you for posting your code!
I think there's a mismatch of the argument order between here and here.
def forward(cls, ctx, input, num_bits=8, min_value=None, max_value=None, stochastic=False, inplace=False, enforce_true_zero=False, num_chunks=None, out_half=False)
UniformQuantize().apply( x, num_bits, min_value, max_value, num_chunks, stochastic, inplace)
Among other potential issues, this causes stochastic arg to take the value of num_chunks, sometimes making it true, and leading to "stochastic" rounding.