yingxin

"out = pool(features, rois)" can not work,the error is the same.but "output = torch.nn.functional.avg_pool2d(features, 7)" can work well.

the full error message is: Using /home/zyd/.cache/torch_extensions as PyTorch extensions root... Detected CUDA files, patching ldflags Emitting ninja build file /home/zyd/.cache/torch_extensions/_prroi_pooling/build.ninja... Building extension module _prroi_pooling... Allowing ninja to set a...