Unet: why the program requires a lot of RAM

[TruongKhang]

Hello,

I am trying to run your source code with the image size 458 x 458 x 3. However, the RAM is required a lot, it is consumed up to 43 GB. Is there any problem here?

Best,
Khang Truong.

[Hsankesara]

Hey @TruongKhang,

Sorry for this super late reply. I was engaged in some of my university stuff and missed the issue altogether. I think it requires memory because of the size of the image and the batch size you're using. Try to decrease it if you want it to take lesser memory. Also, you can use images of smaller dimensions or removing a few-layer can help.