This discussion might be relevant:

  
      github.com/pytorch/pytorch
  
  
    



Issue: [feature request] ROI Pooling as nn layers 31


	
	opened by varunagrawal
	on 2018-01-30
	
	
	


Going through the code, I see there is already support for ROI Pooling CUDA kernels. It would be great to have...