Detectron.pytorch
Detectron.pytorch copied to clipboard
Memory-Usage of GPU increases when training if i use ROIPOOLING
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?