From 8679b7765013ac7f535ca8bc627a0aa5950d01af Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=AD=8F=E9=B9=8F=E9=A3=9E?= Date: Mon, 23 May 2022 11:25:41 +0000 Subject: [PATCH] =?UTF-8?q?ResNet50=5Ffor=5FPyTorch=E6=A8=A1=E5=9E=8B?= =?UTF-8?q?=E4=BC=98=E5=8C=96README.md?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../built-in/cv/classification/ResNet50_for_PyTorch/README.md | 3 +++ 1 file changed, 3 insertions(+) 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 1de5116b9a..aead56f080 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 + 二、训练流程： 单卡训练流程： -- Gitee