From 57c6f8ad24ba9eec810c57d633ae79551dbd841b Mon Sep 17 00:00:00 2001 From: Frank Zhang Date: Thu, 3 Oct 2019 16:04:20 -0400 Subject: [PATCH] Change the GPU policy to wait instead of yes This is needed to avoid failures when using more number jobs than number of GPUs. Previously when --use-gpu=yes, when all GPUs are occupied, the job will error out in 20s. Changing to the policy to wait will have correct behavior. --- egs/wsj/s5/steps/nnet3/align.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/wsj/s5/steps/nnet3/align.sh b/egs/wsj/s5/steps/nnet3/align.sh index aa2de2ee1a5..948d5a4dc38 100755 --- a/egs/wsj/s5/steps/nnet3/align.sh +++ b/egs/wsj/s5/steps/nnet3/align.sh @@ -59,7 +59,7 @@ sdata=$data/split${nj} if $use_gpu; then queue_opt="--gpu 1" - gpu_opt="--use-gpu=yes" + gpu_opt="--use-gpu=wait" else queue_opt="" gpu_opt="--use-gpu=no"