Hi, it was really difficult to figure out the issue without the full stack trace. However, I just took a look at your code and it’s failing in the evaluate function all the way at the bottom. The size of your labels look fine after passing through the DataLoader. It’s failing at self(images). You might want to revisit your EmergencyVehicleDetectionModel class and ensure the size of the inputs and targets match as intended.