Torchvision.utils.save_image cannot save the whole image

Hi,when I use torchvision.utils.save_image for tensor visualization,
it often only saves the upper region of the image.
For instance:

I have tried saving image only on the main process, and also have tried dist.barrier after the saving code.

What do you think is the main problem??

Thanks for reading!