the error in the 124th row of data_loader

The generated image in Figure 4 of the original paper does not correspond to the code in the data loader. The correct code should delete /255 in the 124th row of data_loader. I hope that the author corrects it to avoid more misleading.