Color saturation?

[litingfeng]

Not sure if I should call the problem as color saturation. In my generated images, I found out there's some pixel chunks, like the black area in the image below.

How should I solve this problem? FYI, the original images are 3232, I resized them to 256256, then train the decoder.

Thanks,

[ZhongliangGuo]

just do torch.clamp(tensor, 0, 1)