Skip to content
Snippets Groups Projects
Unverified Commit 2f5db948 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!3528 update tbnet shell scripts (#I5MZ3S)

Merge pull request !3528 from lihaoyang/master
parents d852fcdd 482e2863
No related branches found
No related tags found
No related merge requests found
......@@ -29,16 +29,17 @@ if [ $# == 3 ]; then
DEVICE_TARGET=$3
fi
python preprocess_dataset.py --dataset $DATA_NAME --device_target $DEVICE_TARGET &> scripts/train_standalone_log &
if [ "$DEVICE_TARGET" = "GPU" ];
then
export CUDA_VISIBLE_DEVICES=$2
python preprocess_dataset.py --dataset $DATA_NAME --device_target $DEVICE_TARGET &> scripts/train_standalone_log &&
python train.py --dataset $DATA_NAME --device_target $DEVICE_TARGET --device_id 0 &>> scripts/train_standalone_log &
fi
if [ "$DEVICE_TARGET" = "Ascend" ];
then
export DEVICE_ID=$2
python preprocess_dataset.py --dataset $DATA_NAME --device_target $DEVICE_TARGET &> scripts/train_standalone_log &&
python train.py --dataset $DATA_NAME --device_target $DEVICE_TARGET --device_id $DEVICE_ID &>> scripts/train_standalone_log &
fi
\ No newline at end of file
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