From c3ec258838f25a5dad89f961203e2e5dfb6dd28d Mon Sep 17 00:00:00 2001 From: root Date: Mon, 24 Oct 2022 16:05:15 +0800 Subject: [PATCH] modify sh --- .../SRNTT-l2_ID0272_for_TensorFlow/test/train_full_1p.sh | 3 ++- .../test/train_performance_1p.sh | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/TensorFlow/built-in/cv/detection/SRNTT-l2_ID0272_for_TensorFlow/test/train_full_1p.sh b/TensorFlow/built-in/cv/detection/SRNTT-l2_ID0272_for_TensorFlow/test/train_full_1p.sh index ee31ba899..ab5ab3ecc 100644 --- a/TensorFlow/built-in/cv/detection/SRNTT-l2_ID0272_for_TensorFlow/test/train_full_1p.sh +++ b/TensorFlow/built-in/cv/detection/SRNTT-l2_ID0272_for_TensorFlow/test/train_full_1p.sh @@ -11,7 +11,8 @@ batch_size=9 #网络名称,同目录名称 Network="SRNTT-l2_ID0272_for_TensorFlow" #Device数量,单卡默认为1 -RANK_SIZE=1 +export RANK_SIZE=1 +export RANK_ID=0 #训练epoch,可选 train_epochs=30 #训练step diff --git a/TensorFlow/built-in/cv/detection/SRNTT-l2_ID0272_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/cv/detection/SRNTT-l2_ID0272_for_TensorFlow/test/train_performance_1p.sh index f9b9abbe6..88518348e 100644 --- a/TensorFlow/built-in/cv/detection/SRNTT-l2_ID0272_for_TensorFlow/test/train_performance_1p.sh +++ b/TensorFlow/built-in/cv/detection/SRNTT-l2_ID0272_for_TensorFlow/test/train_performance_1p.sh @@ -11,7 +11,8 @@ batch_size=9 #网络名称,同目录名称 Network="SRNTT-l2_ID0272_for_TensorFlow" #Device数量,单卡默认为1 -RANK_SIZE=1 +export RANK_SIZE=1 +export RANK_ID=0 #训练epoch,可选 train_epochs=30 #训练step -- Gitee