p85620714
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!6773 fix deepfm script error 

Merge pull request !6773 from panfengfeng/fix_deepfm_mnasnet_bug
This commit is contained in:
mindspore-ci-bot 2020-09-24 09:08:31 +08:00 committed by Gitee
parent 8e79826002 8da68ee0b5
commit 2bac83ba1b
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
model_zoo/official/recommend/deepfm/scripts/run_standalone_train.sh
View File

@ -14,12 +14,22 @@
# limitations under the License. # limitations under the License.
# ============================================================================ # ============================================================================
echo "Please run the script as: " echo "Please run the script as: "
echo "sh scripts/run_standalone_train.sh DEVICE_ID DEVICE_TARGET DATASET_PATH" echo "sh scripts/run_standalone_train.sh DEVICE_ID/CUDA_VISIBLE_DEVICES DEVICE_TARGET DATASET_PATH"
echo "for example: sh scripts/run_standalone_train.sh 0 GPU /dataset_path" echo "for example: sh scripts/run_standalone_train.sh 0 GPU /dataset_path"
echo "After running the script, the network runs in the background, The log will be generated in ms_log/output.log" echo "After running the script, the network runs in the background, The log will be generated in ms_log/output.log"
export DEVICE_ID=$1
DEVICE_TARGET=$2 DEVICE_TARGET=$2
if [ "$DEVICE_TARGET" = "GPU" ]
then
export CUDA_VISIBLE_DEVICES=$1
fi
if [ "$DEVICE_TARGET" = "Ascend" ];
then
export DEVICE_ID=$1
fi
DATA_URL=$3 DATA_URL=$3
mkdir -p ms_log mkdir -p ms_log