Looking at it, the only thing it triggers me would be the dataloader...
But if it work with the other models, would work with this too.

Can you share your dataloader code?