How to call RoiPooling2d?

1

Hi,

I’d like to use the RoiPooling2d function implemented here

aten/src/ATen/native/RoiPooling.cpp
aten/src/ATen/native/cuda/RoiPooling.cu

How can I call this from python?

This is not working:
torch._C._VariableBase.RoiPooling2d_forward()

Thanks!

Edit: I built PyTorch from the latest sources from GitHub

2

It’s coming soon. See this issue for more details.

The functions in
aten/src/ATen/native/RoiPooling.cpp
aten/src/ATen/native/cuda/RoiPooling.cu
don’t work too well at the moment.

2 Likes