OOM error occurs
RuntimeError: CUDA out of memory. Tried to allocate 262.00 MiB (GPU 0; 23.70 GiB total capacity; 21.79 GiB already allocated; 20.81 MiB free; 22.32 GiB reserved in total by PyTorch)
when I use command
python prompt/run.py train data/lvis_clip_image_proposal_embedding/train data/lvis_clip_image_proposal_embedding/val checkpoints/exp fg_bg_5_5_6_end soft 0.5 0.5 0.6 8 end
on 3090 (24G).
So I wonder how much memory is needed to train detpro?
OOM error occurs
RuntimeError: CUDA out of memory. Tried to allocate 262.00 MiB (GPU 0; 23.70 GiB total capacity; 21.79 GiB already allocated; 20.81 MiB free; 22.32 GiB reserved in total by PyTorch)
when I use commandpython prompt/run.py train data/lvis_clip_image_proposal_embedding/train data/lvis_clip_image_proposal_embedding/val checkpoints/exp fg_bg_5_5_6_end soft 0.5 0.5 0.6 8 end
on 3090 (24G).So I wonder how much memory is needed to train detpro?