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..06ae28db18d48be9e21c9a7bd5c761cf375c15a9 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 + 二、训练流程: 单卡训练流程: