Torchvision.utils.save_image cannot save the whole image

vision
1

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

스크린샷 2023-02-20 오후 5.29.21
스크린샷 2023-02-20 오후 5.29.211920×650 66.4 KB

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

스크린샷 2023-02-20 오후 5.31.17
스크린샷 2023-02-20 오후 5.31.171368×144 22.1 KB

What do you think is the main problem??

Thanks for reading!