torch.cuda.OutOfMemoryError: CUDA out of memory

longzw1997 / Open-GroundingDino

This is the third party implementation of the paper Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection.

MIT License

386 stars 60 forks source link

torch.cuda.OutOfMemoryError: CUDA out of memory #68

Closed otakue closed 4 months ago

otakue commented 4 months ago

torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 1000.00 MiB. GPU , I have 12g of video memory, This happens when training 100 images, are there any parameters to set?