diff --git a/PyTorch/contrib/cv/classification/MobileNet/test/train_eval_8p.sh b/PyTorch/contrib/cv/classification/MobileNet/test/train_eval_8p.sh index 7601be3ffc78b403a8e24b958d8f56825c56787e..6fdbe525b0d335d696dcebfbc6138dfd0ada7ff4 100755 --- a/PyTorch/contrib/cv/classification/MobileNet/test/train_eval_8p.sh +++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_eval_8p.sh @@ -75,7 +75,7 @@ do KERNEL_NUM=$(($(nproc)/8)) PID_START=$((KERNEL_NUM * RANK_ID)) PID_END=$((PID_START + KERNEL_NUM - 1)) -taskset -c $PID_START-$PID_END python mobilenet.py \ +taskset -c $PID_START-$PID_END python3.7 mobilenet.py \ --data=${data_path} \ -e \ --batch_size=${bach_size} \ 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 9e96f28b95db62e86e44f293edc8c883b7d0a64d..55b11273c94594277be939c7296414ba46b0f582 100755 --- a/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh +++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh @@ -12,7 +12,7 @@ export RANK_SIZE=1 data_path="" # 训练epoch -train_epochs=1 +train_epochs=90 # 指定训练所使用的npu device卡id device_id=0 # 学习率 diff --git a/PyTorch/contrib/cv/classification/MobileNet/test/train_full_8p.sh b/PyTorch/contrib/cv/classification/MobileNet/test/train_full_8p.sh index c2bf7064cc484707697a142db4755771852216a6..41889c650f6bf15f2449b8bdd14eaeef265d86c9 100755 --- a/PyTorch/contrib/cv/classification/MobileNet/test/train_full_8p.sh +++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_full_8p.sh @@ -75,7 +75,7 @@ KERNEL_NUM=$(($(nproc)/8)) PID_START=$((KERNEL_NUM * RANK_ID)) PID_END=$((PID_START + KERNEL_NUM - 1)) -taskset -c $PID_START-$PID_END python mobilenet.py \ +taskset -c $PID_START-$PID_END python3.7 mobilenet.py \ --data=${data_path} \ --batch_size=${batch_size} \ --ngpu 8 \ 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 0d21e2cdd76f966f2a36411121c77a23446b538b..b790357f847dcec3d97783c56b4bf48225d5ea51 100755 --- a/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh +++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_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_8p.sh b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh index 8ab1d2e097826331ef4a0e68e1d80fdac7ef9ed4..0d8120807b069ced2135f8becf9a03b4e570d04b 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=90 +train_epochs=1 # 学习率 learning_rate=0.8 @@ -75,7 +75,7 @@ KERNEL_NUM=$(($(nproc)/8)) PID_START=$((KERNEL_NUM * RANK_ID)) PID_END=$((PID_START + KERNEL_NUM - 1)) -taskset -c $PID_START-$PID_END python mobilenet.py \ +taskset -c $PID_START-$PID_END python3.7 mobilenet.py \ --data=${data_path} \ --batch_size=${batch_size} \ --ngpu 8 \