From f62d6314b93416b301ca7bb0b62ccf9c4e6933e2 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 11 七月 2023 00:52:21 +0800
Subject: [PATCH] update
---
egs/aishell/branchformer/run.sh | 9 ++++++---
1 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/egs/aishell/branchformer/run.sh b/egs/aishell/branchformer/run.sh
index b5aeae7..37336ea 100755
--- a/egs/aishell/branchformer/run.sh
+++ b/egs/aishell/branchformer/run.sh
@@ -3,8 +3,8 @@
. ./path.sh || exit 1;
# machines configuration
-CUDA_VISIBLE_DEVICES="0,1"
-gpu_num=2
+CUDA_VISIBLE_DEVICES="0,1,2,3"
+gpu_num=4
count=1
gpu_inference=true # Whether to perform gpu decoding, set false for cpu decoding
# for gpu decoding, inference_nj=ngpu*njob; for cpu decoding, inference_nj=njob
@@ -46,7 +46,10 @@
valid_set=dev
test_sets="dev test"
-asr_config=conf/train_asr_branchformer.yaml
+#asr_config=conf/train_asr_branchformer.yaml
+#asr_config=conf/train_asr_branchformer_bs16000.yaml
+asr_config=conf/train_asr_branchformer_bs16000_gc2.yaml
+#asr_config=conf/train_asr_branchformer_bs10000.yaml
model_dir="baseline_$(basename "${asr_config}" .yaml)_${lang}_${token_type}_${tag}"
inference_config=conf/decode_asr_transformer.yaml
--
Gitblit v1.9.1