Cuda out of memory when training D2Det with iSAID dateset

JialeCao001 / D2Det

D2Det: Towards High Quality Object Detection and Instance Segmentation (CVPR2020)

https://openaccess.thecvf.com/content_CVPR_2020/papers/Cao_D2Det_Towards_High_Quality_Object_Detection_and_Instance_Segmentation_CVPR_2020_paper.pdf

MIT License

297 stars 86 forks source link

Cuda out of memory when training D2Det with iSAID dateset #42

Open jinglingzhucuolesenlin opened 3 years ago

jinglingzhucuolesenlin commented 3 years ago

Excuse me, during my training of your D2Det instance model with iSAID dataset, cuda out of memory at 800 iterations. I cropped the original images into 800×800 pixels with stride 200 and used 1 Titan GPU with a batchsize of 2. Could you please give me some advice ?