Yes, the given code runs, however, when I change the dataset back to my original dataset, the error occurs. My guess is that it has to do smth with uneven number of bounding boxes per image, since that’s the only unusual thing that I can point to. Since some images can have 100 bounding boxes, some 300, etc…