Jdway

OOM Error with `roi_align` in PyTorch 2.1.1 but fine in PyTorch 2.0.1

17

### 🐛 Describe the bug ## Description I am encountering an Out of Memory (OOM) error when using the roi_align function from PyTorch version 2.1.1 with torchvision 0.16.1. This issue...

Thank you for your hard work on this project! I've noticed that the license information for this repository isn't specified. I am familiar with your other project, YOWOv2, which uses...