From b81396f912456a6a9d3813467242a805e7ce7553 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期一, 01 四月 2024 14:31:48 +0800
Subject: [PATCH] add dec reset for batch
---
examples/aishell/transformer/run.sh | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/examples/aishell/transformer/run.sh b/examples/aishell/transformer/run.sh
index 12aef53..575abf8 100755
--- a/examples/aishell/transformer/run.sh
+++ b/examples/aishell/transformer/run.sh
@@ -5,7 +5,7 @@
# general configuration
feats_dir="../DATA" #feature output dictionary
-exp_dir="."
+exp_dir=`pwd`
lang=zh
token_type=char
stage=0
@@ -174,6 +174,7 @@
++output_dir="${inference_dir}/${JOB}" \
++device="${inference_device}" \
++ncpu=1 \
+ ++disable_log=true \
++batch_size="${inference_batch_size}" &> ${_logdir}/log.${JOB}.txt
}&
--
Gitblit v1.9.1