Skip to content

Commit 3a541d7

Browse files
authored
Merge pull request #59 from MooreThreads/wangkang-mt-patch-1
Update run_dist_train.sh
2 parents ce1b4ab + 162f687 commit 3a541d7

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pytorch/nlp/bert/run_dist_train.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
start_time=$(date +%s)
33

44
cur_path=`pwd`
5+
mkdir -p "$cur_path/log"
6+
57
export AMP_ON=1
68
export DDP_ON=1
79
export DEVICE='musa'
@@ -13,4 +15,4 @@ python -m torch.distributed.run --nproc_per_node 8 examples/sequence_labeling/ta
1315
wait
1416
end_time=$(date +%s)
1517
e2e_time=$(( $end_time - $start_time ))
16-
echo "end to end time: $e2e_time" >> ttime.log
18+
echo "end to end time: $e2e_time" >> ttime.log

0 commit comments

Comments
 (0)