RegTR icon indicating copy to clipboard operation
RegTR copied to clipboard

Published 20 hours ago • verified publisher icon yewzijian

CUDA out of memory

Open lcxiha opened this issue 1 year ago • 0 comments

Hi. When using the model you trained to validate my dataset in demo. py and train my own dataset , the same error occurred: CUDA out of memory .Tried to allocate 874.00 MiB(GPU 0; 7.79 GiB total capacity; 3.85GiB already allocated; 695.06GiB free; 5.04 GiB reserved in total by PyTorch) .My dataset only has 17 pairs of point clouds, and I have reduced the batch_size and base_ Lr still has this error. Is there any other way besides replacing the graphics card. Looking forward to your reply.

lcxiha avatar Jun 09 '23 03:06 lcxiha