Open anuragarnab opened 2 years ago
cc @fuqichen1998
Have you confirmed that this change will affect torchvision versions > 0.1.0? If not, we prefer to keep the current code, and you could report this as an issue.
Yes. This correctly parses the version number for older torchvision versions too.
The current check fails for torchvision versions >= 0.10. This commit corrects the parsing of the version number.