I find that the function roi_align causes this error, so I want how you can use this function and do not have an error? My environment is: torch-1.4, torchvision-0.5. And other types of torch and torchvision I have tried but do not work. In your requirement torch is 1.5 version and torchvision is 0.5 version, but they are not matching.