How to build ROIAlign for pytorch 0.4
MathewSam opened this issue · 0 comments
MathewSam commented
Do we have to do the same operations for both pytorch 0.3.1 and pytorch 0.4 to build ROIAlign? The demo file keeps restarting every time I pass a sample image to the model and this is the only thing I am doubtful about.
Thanks