@dskhudia
Thank you very much, I changed
score_link = y[0,:,:,1].cpu().data.numpy() to score_link = y[0,:,:,1].int_repr().cpu().data.numpy()  as per your suggestion. But the prediction is very bad.
Can you point me to how to dequantize the model ?