Thanks for the replies. I’m already using the master build so I’ll try it out. I have an image segmentation model to test, it always uncovers issues that folks working with imagenet models do not see.
@jerryzh168 Are you open for contribution? I’ve never contributed to such a large project as PyTorch, but I’m interested in the opportunity.