vacancy / PreciseRoIPooling

Precise RoI Pooling with coordinate gradient support, proposed in the paper "Acquisition of Localization Confidence for Accurate Object Detection" (https://arxiv.org/abs/1807.11590).
MIT License
770 stars 152 forks source link

pytorch1.0 compile #54

Closed laisimiao closed 3 years ago

laisimiao commented 3 years ago

So under pytorch1.0, it seems once you run code and it will re-compile PrPooling module, is there any way to use compiled module and doesn't need re-compile anymore in future run?

vacancy commented 3 years ago

PyTorch should do the cache automatically.