diff --git a/PyTorch/built-in/cv/classification/ResNet50_for_PyTorch/README.md b/PyTorch/built-in/cv/classification/ResNet50_for_PyTorch/README.md index 1de5116b9a4a6f26341c803f81393baa06b93d15..aead56f080f32737fb81361bc4cba99cfed1760a 100644 --- a/PyTorch/built-in/cv/classification/ResNet50_for_PyTorch/README.md +++ b/PyTorch/built-in/cv/classification/ResNet50_for_PyTorch/README.md @@ -7,6 +7,9 @@ * torchvision * dllogger +注:pillow建议安装较新版本, 与之对应的torchvision版本如果无法直接安装,可使用源码安装对应的版本,源码参考链接:https://github.com/pytorch/vision; +建议:Pillow版本是9.1.0,torchvision版本x86平台是0.6.0, arm平台是0.6.0a0 + 二、训练流程: 单卡训练流程: