diff --git a/official/recommend/tbnet/scripts/run_standalone_train.sh b/official/recommend/tbnet/scripts/run_standalone_train.sh index aaa8fc6d94dcd59cf77b1cc034632bc9de4cfc8b..fbd6333246ed616862f7facc327e0b235d46a03c 100644 --- a/official/recommend/tbnet/scripts/run_standalone_train.sh +++ b/official/recommend/tbnet/scripts/run_standalone_train.sh @@ -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