running wage_quantizer causes an issue

[Saadmaghani]

QG function takes in 4 inputs on line 47 of models.wage_quantizer.py but is given 5 in the testing function on line 128.

def QG(x, bits_G, bits_R, lr):

quant_data = QG(test_tensor, bits_G, bits_R, lr, rand_tensor).data.numpy()