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

fix deepfm gpu train script

This commit is contained in:
panfengfeng 2020-09-23 14:12:42 +08:00
parent 7f35e2e2a1
commit 8da68ee0b5
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.
# ============================================================================
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 "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
if [ "$DEVICE_TARGET" = "GPU" ]
then
export CUDA_VISIBLE_DEVICES=$1
fi
if [ "$DEVICE_TARGET" = "Ascend" ];
then
export DEVICE_ID=$1
fi
DATA_URL=$3
mkdir -p ms_log