不幸遇到了如下的警告
/home/chauncey/.local/lib/python3.8/site-packages/torchvision/transforms/functional.py:364: UserWarning: Argument interpolation should be of type InterpolationMode instead of int. Please, use InterpolationMode enum. warnings.warn(
有些解释是用InterpolationMode.BICUBIC
替代 Image.BICUBIC
参考 [pytroch warning]
但是,这其实是torchvision和pillow不兼容导致的,我的设备里torchvision=0.9.1 and pillow=7.0.0,即使我把pillow升级到8.3.1,依然有warning。
那只能降低torchvision了,但是torchvision的版本号一般都是和pytorch绑定好的,我们需要不依赖torch来更改torchvison的版本,这可以通过以下指令实现 参考解决安装torchvision自动更新torch到最新版本
pip3 install --no-deps torchvision==0.8.2
再将pillow降到6.2.2即可,其他版本的pillow就没有尝试过了
感谢changlin31的指导