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

!616 [Bug Fix] Issue I4G4NC fix gpu train scripts

Merge pull request !616 from ZeyangGAO/fix_glore_res
parents 0e78850b c0266a38
No related branches found
No related tags found
No related merge requests found
......@@ -55,7 +55,7 @@ cp src/*.py ./train_parallel/src
cd ./train_parallel
env > env.log
echo "start training"
mpirun -n $2 --allow-run-as-root \
mpirun -n $3 --allow-run-as-root \
python3 train.py --data_url=$DATA_PATH --isModelArts=False --run_distribute=True \
--device_target="GPU" --config_path=$CONFIG_PATH --eval_data_url=$EVAL_DATA_PATH --device_num $2 > train.log 2>&1 &
--device_target="GPU" --config_path=$CONFIG_PATH --eval_data_url=$EVAL_DATA_PATH --device_num $3 > 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