Detectron.pytorch icon indicating copy to clipboard operation
Detectron.pytorch copied to clipboard

Memory-Usage of GPU increases when training if i use ROIPOOLING

Open GYxiaOH opened this issue 6 years ago • 0 comments

i find this code only can use roialign ,if i use roipooling ,Memory-Usage of GPU increases when training until out of memory. i add some new roi such as psroi ,and i meet same problem, i don't know why, Has anyone else encountered this problem?

GYxiaOH avatar Aug 20 '18 09:08 GYxiaOH