From f7f1d7a43c1d13c95a42f8f5b11d156e29ad7747 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Tue, 12 Apr 2022 11:42:23 +0000 Subject: [PATCH 1/4] update PyTorch/contrib/cv/classification/WideResNet101_2_for_Pytorch/test/train_performance_1p.sh. --- .../WideResNet101_2_for_Pytorch/test/train_performance_1p.sh | 1 + 1 file changed, 1 insertion(+) 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 -- Gitee From 887d3aff9f65d2536162e85caac1d346c1c463ef Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Tue, 12 Apr 2022 11:44:43 +0000 Subject: [PATCH 2/4] update PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh. --- .../cv/classification/MobileNet/test/train_performance_1p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 # 学习率 -- Gitee From 2c75ce9541697845a82c3b0295f64edc74f990fc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Tue, 12 Apr 2022 11:45:14 +0000 Subject: [PATCH 3/4] update PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh. --- .../contrib/cv/classification/MobileNet/test/train_full_1p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 -- Gitee From 03d13ab51fcc7024d630321ee8caf51b2de4c062 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Tue, 12 Apr 2022 11:45:35 +0000 Subject: [PATCH 4/4] update PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh. --- .../cv/classification/MobileNet/test/train_performance_8p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 -- Gitee