diff --git a/cv/classification/resnet50/pytorch/scripts/amp_1card.sh b/cv/classification/resnet50/pytorch/scripts/amp_1card.sh index d79c56c32c15c9bef45425c802de63b3278337d3..10bb9f59716d28f1d71b4eacfe14b79c3f69e7cb 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 85e477fd93df9f2881a9dc660ed2437f26d72fe5..679873ef9243d746c50ac0c70177cd70e609dd58 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 ad8fb57922d253577c49b37edb0ac0031967f460..7a398ed87b65eda145272d93eeb81b03ab165b14 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 $@