search

joe-siyuan-qiao / DetectoRS

DetectoRS: Detecting Objects with Recursive Feature Pyramid and Switchable Atrous Convolution
Apache License 2.0
1.13k stars 178 forks source link

GPU and training #85

Open Yurushia1998 opened 3 years ago

Yurushia1998 commented 3 years ago

I'm testing DetectoRS training using only 1 GPU with 1 image per gpu, 1 worker per gpu and the cuda memory is full already. My GPU has 11GB in memory. How comes a detection models can take this much memory, or is there a chance that I do anything wrong? What is the requirement Cuda memory per gpu for this models?

Keysmis commented 3 years ago

@Yurushia1998 have you solve this problem? i meet the same, the gpu memory is too high~