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 87205b560a989fb3f6c074aab08178403bff9d5b..138c06d613df1c4d93e0e16a24fc8f02394dca80 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 &