From 1af68ba6ffc21d4dc3bd5f01cda656def97e361c Mon Sep 17 00:00:00 2001
From: Nixon <2465004358@qq.com>
Date: 星期六, 14 九月 2024 10:13:23 +0800
Subject: [PATCH] fix bug, 1 fix cuda oom, 2 fix choose a window size 400 that is [2, 0] (#2075)

---
 examples/aishell/branchformer/run.sh |    3 +++
 1 files changed, 3 insertions(+), 0 deletions(-)

diff --git a/examples/aishell/branchformer/run.sh b/examples/aishell/branchformer/run.sh
index 918aa9b..5b64954 100755
--- a/examples/aishell/branchformer/run.sh
+++ b/examples/aishell/branchformer/run.sh
@@ -27,6 +27,8 @@
 tag="exp1"
 workspace=`pwd`
 
+master_port=12345
+
 . utils/parse_options.sh || exit 1;
 
 # Set bash to 'debug' mode, it will exit on :
@@ -115,6 +117,7 @@
   torchrun \
   --nnodes 1 \
   --nproc_per_node ${gpu_num} \
+  --master_port ${master_port} \
   ../../../funasr/bin/train.py \
   --config-path "${workspace}/conf" \
   --config-name "${config}" \

--
Gitblit v1.9.1