search

facebookresearch / VICRegL

VICRegL official code base
Other
223 stars 24 forks source link

No 'InterpolationMode' in torchvision 0.8.2 #7

Closed hugoych closed 1 year ago

hugoych commented 1 year ago

Hi, See the previous issue from the vicreg repo (https://github.com/facebookresearch/vicreg/issues/19).

If I'm correct, torchvision.transforms.InterpolationMode has only been introduced in 0.9.0 version. Thus, the requirement mentioned in the README of torchvision>=0.8.2 can lead to an ImportError.

ds2268 commented 1 year ago

I have tested it on the latest PyTorch and it works, with a small modification - https://github.com/facebookresearch/VICRegL/pull/8.

Adrien987k commented 1 year ago

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.