search

zhuye98 / ICL

The official repo for [MIDL'23 Oral] "Inherent Consistent Learning for Accurate Semi-supervised Medical Image Segmentation"
MIT License
26 stars 3 forks source link

cuda out of memory #2

Closed pengluxy closed 1 year ago

pengluxy commented 1 year ago

Hello author, this paper you wrote is very inspiring! Thank you and your team. But when I reproduce the thesis, there is always insufficient graphics card memory. I am very confused. My GPU has 12G. I have tried various methods, but it still doesn't work. What is the reason?

zhuye98 commented 1 year ago

Hi pengluxy, this problem may occur when the GPU memory is too small. It is recommended that the GPU memory is at least larger than 32G to train the model.

pengluxy commented 1 year ago

okok! thanks a lot!