We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents ce1b4ab + 162f687 commit 3a541d7Copy full SHA for 3a541d7
pytorch/nlp/bert/run_dist_train.sh
@@ -2,6 +2,8 @@
2
start_time=$(date +%s)
3
4
cur_path=`pwd`
5
+mkdir -p "$cur_path/log"
6
+
7
export AMP_ON=1
8
export DDP_ON=1
9
export DEVICE='musa'
@@ -13,4 +15,4 @@ python -m torch.distributed.run --nproc_per_node 8 examples/sequence_labeling/ta
13
15
wait
14
16
end_time=$(date +%s)
17
e2e_time=$(( $end_time - $start_time ))
-echo "end to end time: $e2e_time" >> ttime.log
18
+echo "end to end time: $e2e_time" >> ttime.log
0 commit comments