That’s right and a good idea to solve the original issue. I would start with making sure the torchvision implementation yields the same results first as a smoke test, as I guess there might be a misunderstanding how seeding works or another issue.