torchvision dependencies mismatch with 'InterpolationMode' class

[hugoych]

Hi,
Thanks for the repo!

I saw from the README that the only requirement was pytorch==1.7.1 but I think there is also an implicit requirement on the vision library: torchvision >=0.9.0.

Indeed from pytorch official installation guide pytorch 1.7.1 should be installed with torchvision 0.8.2 (see doc). However, according to the torchvision repo, the class InterpolationMode, only appears in version 0.9.0.

This leads to an ImportError in augmentations.py.

Do you know the reason behind this and can you confirm which version of torchvision you are using?

Thanks!

[Adrien987k]

Hi,

Thanks for pointing this out! The correct pytorch version is actually 1.8.1 and torchvision is 0.9.1.
I have updated the readme.