From 798ab06e19a88b723678356106ea2792cb53868b Mon Sep 17 00:00:00 2001
From: panfengfeng <panfengfeng@huawei.com>
Date: Wed, 1 Dec 2021 04:09:38 +0800
Subject: [PATCH] fix nasnet 1p running

---
 official/cv/nasnet/scripts/run_standalone_train_for_gpu.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/official/cv/nasnet/scripts/run_standalone_train_for_gpu.sh b/official/cv/nasnet/scripts/run_standalone_train_for_gpu.sh
index 87205b560..138c06d61 100644
--- a/official/cv/nasnet/scripts/run_standalone_train_for_gpu.sh
+++ b/official/cv/nasnet/scripts/run_standalone_train_for_gpu.sh
@@ -41,5 +41,5 @@ fi
 
 DEVICE_ID=$1
 DATASET_PATH=$2
-CUDA_VISIBLE_DEVICES=$DEVICE_ID python ../train.py --platform='GPU' --device_id=$DEVICE_ID --dataset_path=$DATASET_PATH > train.log 2>&1 &
+CUDA_VISIBLE_DEVICES=$DEVICE_ID python ../train.py --platform='GPU' --dataset_path=$DATASET_PATH > train.log 2>&1 &
 
-- 
GitLab