compare_roi_pooling_speed
compare_roi_pooling_speed copied to clipboard
CUDA: OUT OF MEMORY
I tried to use your roi_pooling code as a layer in my model. But every time it iterates increasing the memory, do you have any idea how to fix this?