Some of the weights become very large when I run the gpu version. There may be some problems in "quantize_weight_backward".
may0324
