I’d have to go back and double check the dates of the various versions to be sure, at the time I was using the latest stable build of both pytorch and pytorch vision. But, I believe this was using pytorch 1.5 and pytorch vision .4?