About stpls3d training error CUDA out of memory about isbnet HOT 6 CLOSED

Hi, I met the same problem before. I use the cloud GPU, so I can change the GPU easily. I used A40(48GB) * 1, the problems have been already solved.

from isbnet.

Hi, I met the same problem before. I use the cloud GPU, so I can change the GPU easily. I used A40(48GB) * 1, the problems have been already solved.

Ok, thanks for your reply, I will try the cloud GPU.

from isbnet.

You can solve this problem by modifying the batch_size<=16. I used 1, then make it.

from isbnet.

I encountered the same problem. But if I don't use cloud GPU, how should I solve this problem?

from isbnet.

I encountered the same problem. But if I don't use cloud GPU, how should I solve this problem?

Due to my poor hardware conditions, I only took the solution proposed by TobyZhouWei and used the cloud GPU to solve it.

from isbnet.

I encountered the same problem. But if I don't use cloud GPU, how should I solve this problem?

You can solve this problem by decreasing the number of batch_size. As for me, I set the batch_size=1.

from isbnet.