Issue
Previous method of directly compaing the third value of the torchvision versioning torchvision.__version__[:3] fails once the version gets above 0.9.0 as 0.1.0 and 0.10.0 get treated as the same value.
Solution
Use the whole segment between the first and second decimal point to get the subversion such that 0.1.0 and 0.10.0 are now different
Possible Issues
This solution will no longer work if torchvision increases its major version up to 1..
Issue Previous method of directly compaing the third value of the torchvision versioning
torchvision.__version__[:3]fails once the version gets above 0.9.0 as 0.1.0 and 0.10.0 get treated as the same value.Solution Use the whole segment between the first and second decimal point to get the subversion such that 0.1.0 and 0.10.0 are now different
Possible Issues This solution will no longer work if torchvision increases its major version up to 1..