From 47390b8f53a32eebfe41424af2dc1bc1033786f3 Mon Sep 17 00:00:00 2001 From: "song.jian" Date: Mon, 5 Dec 2022 14:15:46 +0800 Subject: [PATCH] batch_size cong 512 xiugaidao 256 Signed-off-by: song.jian --- cv/classification/resnest50/pytorch/train_resnest50_amp_dist.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cv/classification/resnest50/pytorch/train_resnest50_amp_dist.sh b/cv/classification/resnest50/pytorch/train_resnest50_amp_dist.sh index af76da2a5..3331e037e 100755 --- a/cv/classification/resnest50/pytorch/train_resnest50_amp_dist.sh +++ b/cv/classification/resnest50/pytorch/train_resnest50_amp_dist.sh @@ -26,5 +26,5 @@ fi cd ${ROOT_DIR} python3 $PYTHONARG ${ROOT_DIR}/run_train.py \ --model resnest50 --dali --dali-cpu --data-path $DATA_PATH \ - --opt fused_sgd --batch-size 512 --lr 0.0125 \ + --opt fused_sgd --batch-size 256 --lr 0.0125 \ --amp --nhwc "$@" -- Gitee