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..99335507d7712c765d377bce5a54763ee211194e 100644 --- a/PyTorch/built-in/cv/classification/ResNet50_for_PyTorch/README.md +++ b/PyTorch/built-in/cv/classification/ResNet50_for_PyTorch/README.md @@ -6,6 +6,9 @@ * torch(NPU版本) * torchvision * dllogger + +注:pillow建议安装较新版本, 与之对应的torchvision版本如果无法直接安装,可使用源码安装对应的版本,源码参考链接:https://github.com/pytorch/vision + 建议:Pillow版本是9.1.0 torchvision版本x86平台是0.6.0, arm平台是0.6.0a0 二、训练流程: