CUDA out of memory when set net.use_mask = True at testing time

uci-cbcl / NoduleNet

[MICCAI' 19] NoduleNet: Decoupled False Positive Reduction for Pulmonary Nodule Detection and Segmentation

Other

188 stars 58 forks source link

Open LXYTSOS opened 4 years ago

LXYTSOS commented 4 years ago

no matter how many GPUs(11GB each) I use, I always get out of memory error when set net.use_mask = True at testing time, how can I solve this?

fengz63 commented 4 years ago

Hi, I also have the same question, Do you have solved it?

LXYTSOS commented 4 years ago

no, I just give up.

fengz63 commented 4 years ago

OK, thank you.

dreamwish1998 commented 3 years ago

no matter how many GPUs(11GB each) I use, I always get out of memory error when set net.use_mask = True at testing time, how can I solve this?

@LXYTSOS What is the GPU usage rate when train.py is running? How about the GPU memory usage?

chovvvvvy commented 1 year ago

@dreamwish1998 Hi, I also have the same question, Do you have solved it？