RuntimeError: CUDA error: out of memory

sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset

MIT License

223 stars 75 forks source link

Closed samahwaleed closed 1 year ago

samahwaleed commented 1 year ago

I want to train the model, but I got this error even the batch size = 1