Skip to content
Snippets Groups Projects
Commit 798ab06e authored by panfengfeng's avatar panfengfeng
Browse files

fix nasnet 1p running

parent fafd17ed
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
official/cv/nasnet/scripts/run_standalone_train_for_gpu.sh
+ 1
− 1
View file @ 798ab06e
......@@ -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 &
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment