From 4a168ae794f7254d18f4dd8daaa76b126e93c41a Mon Sep 17 00:00:00 2001 From: bailang Date: Thu, 14 Apr 2022 10:56:20 +0800 Subject: [PATCH] =?UTF-8?q?[=E4=BC=97=E6=99=BA][Pytorch]=E4=BC=97=E6=99=BA?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E8=84=9A=E6=9C=AC=E5=BE=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: bailang --- .../contrib/cv/classification/MobileNet/test/train_full_1p.sh | 2 +- .../cv/classification/MobileNet/test/train_performance_1p.sh | 2 +- .../cv/classification/MobileNet/test/train_performance_8p.sh | 2 +- .../WideResNet101_2_for_Pytorch/test/train_performance_1p.sh | 1 + PyTorch/contrib/cv/detection/SSD/test/train_performance_8p.sh | 2 +- 5 files changed, 5 insertions(+), 4 deletions(-) diff --git a/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh b/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh index 55b11273c9..ff208ef246 100755 --- a/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh +++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh @@ -85,7 +85,7 @@ python3.7 mobilenet.py \ --epochs=${train_epochs} \ -j $(($(nproc))) \ --lr=${learning_rate} \ - --device_id={ASCEND_DEVICE_ID} ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & + --device_id=${ASCEND_DEVICE_ID} ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & wait diff --git a/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh index b790357f84..5ab530c3c0 100755 --- a/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh +++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh @@ -12,7 +12,7 @@ export RANK_SIZE=1 data_path="" # 训练epoch -train_epochs=1 +train_epochs=2 # 指定训练所使用的npu device卡id device_id=0 # 学习率 diff --git a/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh index 0d8120807b..373b7a034d 100755 --- a/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh +++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh @@ -12,7 +12,7 @@ export RANK_SIZE=8 data_path="" # 训练epoch -train_epochs=1 +train_epochs=2 # 学习率 learning_rate=0.8 diff --git a/PyTorch/contrib/cv/classification/WideResNet101_2_for_Pytorch/test/train_performance_1p.sh b/PyTorch/contrib/cv/classification/WideResNet101_2_for_Pytorch/test/train_performance_1p.sh index a05d07a366..0d05811ca3 100644 --- a/PyTorch/contrib/cv/classification/WideResNet101_2_for_Pytorch/test/train_performance_1p.sh +++ b/PyTorch/contrib/cv/classification/WideResNet101_2_for_Pytorch/test/train_performance_1p.sh @@ -116,6 +116,7 @@ python3.7 -u ./main_npu_1p.py \ --opt-level='O2' \ --device='npu' \ --world-size=1 \ + --npu=${ASCEND_DEVICE_ID} \ --batch-size=${batch_size} > ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & wait diff --git a/PyTorch/contrib/cv/detection/SSD/test/train_performance_8p.sh b/PyTorch/contrib/cv/detection/SSD/test/train_performance_8p.sh index d265302391..ae08912df7 100644 --- a/PyTorch/contrib/cv/detection/SSD/test/train_performance_8p.sh +++ b/PyTorch/contrib/cv/detection/SSD/test/train_performance_8p.sh @@ -12,7 +12,7 @@ export RANK_SIZE=8 data_path="" # 训练epoch, 默认值24 -train_epochs=24 +train_epochs=2 # 参数校验，data_path为必传参数，其他参数的增删由模型自身决定；此处新增参数需在上面有定义并赋值 -- Gitee