From c00f5f498c8892172b5748e11b9844ce51e54037 Mon Sep 17 00:00:00 2001 From: "yufei.chen" Date: Thu, 27 Oct 2022 14:39:06 +0800 Subject: [PATCH 1/2] modify default batchsize on vgg link #I5Y018 modify default batchsize on vgg Signed-off-by: yufei.chen --- cv/classification/vgg/pytorch/train_vgg16_amp_dist.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cv/classification/vgg/pytorch/train_vgg16_amp_dist.sh b/cv/classification/vgg/pytorch/train_vgg16_amp_dist.sh index 68e3ff187..4fa6953da 100755 --- a/cv/classification/vgg/pytorch/train_vgg16_amp_dist.sh +++ b/cv/classification/vgg/pytorch/train_vgg16_amp_dist.sh @@ -26,5 +26,5 @@ fi cd ${ROOT_DIR} python3 $PYTHONARG ${ROOT_DIR}/run_train.py \ --model vgg16 --dali --dali-cpu --data-path $DATA_PATH \ - --opt fused_sgd --batch-size 300 --lr 1e-2 \ + --opt fused_sgd --batch-size 200 --lr 1e-2 \ --amp --nhwc "$@" -- Gitee From 8d9cbb00d282065bfa66bd17686c7cdf3ee68117 Mon Sep 17 00:00:00 2001 From: "yufei.chen" Date: Thu, 27 Oct 2022 14:55:39 +0800 Subject: [PATCH 2/2] modify README.md with resent50 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit link #I5Y012 resnet50 README.md 更新pip安装包 Signed-off-by: yufei.chen --- cv/classification/resnet50/pytorch/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cv/classification/resnet50/pytorch/README.md b/cv/classification/resnet50/pytorch/README.md index 6e4e5da38..5a4f9318c 100644 --- a/cv/classification/resnet50/pytorch/README.md +++ b/cv/classification/resnet50/pytorch/README.md @@ -4,7 +4,7 @@ Residual Networks, or ResNets, learn residual functions with reference to the la ## Step 1: Installing ```bash -pip3 install -r requirements.txt +pip3 install torch torchvision ``` :beers: Done! -- Gitee