From 0fb0c2097bcc928613f3af222d73b63c4026306c Mon Sep 17 00:00:00 2001 From: "hongliang.yuan" Date: Fri, 3 Jan 2025 11:04:26 +0800 Subject: [PATCH] fix resnet50 script error --- cv/classification/resnet50/pytorch/scripts/amp_1card.sh | 2 +- cv/classification/resnet50/pytorch/scripts/amp_4cards.sh | 2 +- cv/classification/resnet50/pytorch/scripts/amp_8cards.sh | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/cv/classification/resnet50/pytorch/scripts/amp_1card.sh b/cv/classification/resnet50/pytorch/scripts/amp_1card.sh index d79c56c32..10bb9f597 100644 --- a/cv/classification/resnet50/pytorch/scripts/amp_1card.sh +++ b/cv/classification/resnet50/pytorch/scripts/amp_1card.sh @@ -9,4 +9,4 @@ SCRIPT_DIR=$(cd `dirname $0`; pwd) PROJECT_DIR=$SCRIPT_DIR/.. export CUDA_VISIBLE_DEVICES=0 -bash $SCRIPT_DIR/train_resnet50_imagenet.sh --batch-size 300 --amp --channels-last --dali $@ +bash $SCRIPT_DIR/train_resnet50_imagenet.sh --batch-size 300 --amp --dali $@ diff --git a/cv/classification/resnet50/pytorch/scripts/amp_4cards.sh b/cv/classification/resnet50/pytorch/scripts/amp_4cards.sh index 85e477fd9..679873ef9 100644 --- a/cv/classification/resnet50/pytorch/scripts/amp_4cards.sh +++ b/cv/classification/resnet50/pytorch/scripts/amp_4cards.sh @@ -11,4 +11,4 @@ PROJECT_DIR=$SCRIPT_DIR/.. export MASTER_ADDR=127.0.0.1 export MASTER_PORT=1001 export CUDA_VISIBLE_DEVICES=0,1,2,3 -bash $SCRIPT_DIR/train_resnet50_imagenet_dist_1x4_torch.sh --batch-size 512 --amp --channels-last --dali $@ +bash $SCRIPT_DIR/train_resnet50_imagenet_dist_1x4_torch.sh --batch-size 512 --amp --dali $@ diff --git a/cv/classification/resnet50/pytorch/scripts/amp_8cards.sh b/cv/classification/resnet50/pytorch/scripts/amp_8cards.sh index ad8fb5792..7a398ed87 100644 --- a/cv/classification/resnet50/pytorch/scripts/amp_8cards.sh +++ b/cv/classification/resnet50/pytorch/scripts/amp_8cards.sh @@ -11,4 +11,4 @@ PROJECT_DIR=$SCRIPT_DIR/.. export MASTER_ADDR=127.0.0.1 export MASTER_PORT=1001 export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 -bash $SCRIPT_DIR/train_resnet50_imagenet_dist_1x4_torch.sh --batch-size 480 --amp --channels-last --dali $@ +bash $SCRIPT_DIR/train_resnet50_imagenet_dist_1x4_torch.sh --batch-size 480 --amp --dali $@ -- Gitee