RuntimeError: CUDA error: out of memory

adityaarun1 / Detectron.pytorch

A PyTorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available. Supports PyTorch 1.1+ and TorchVision 0.3.0+. No compilation necessary.

MIT License

65 stars 18 forks source link

RuntimeError: CUDA error: out of memory #11

Closed rmxhhh closed 4 years ago

adityaarun1 commented 4 years ago

Can you provide some more information, like the log file? Seems like you don't have enough GPU memory to run the code.

adityaarun1 commented 4 years ago

Closing issue due to lack of response.