Thanks for the stack trace. I don’t know, which operation fails exactly in roi_align, but guess that the target tensor might contain invalid indices.
Could you post the images and targets as well as the model for the failed operation so that we could try to reproduce and debug it?