CUDA Out of memory

[li-Alice-li]

Dear authors,

Thank you very much for your contribution, I am very interested in this work of yours. However, I'm having some problems reproducing the code. I am using a graphics card rtx4090 and the environment is configured according to the parameters you gave. Running the programme prompts the error, "cuda out of memory". Then, I set crop_size=(1024,1024) to (512,512) for the programme to run. The segmentation accuracy obtained is slightly lower. This is my training log, how can I fix the memory problem on rtx4090?

Thanks and best regards
20240619_222500.log