diff --git a/examples/aishell/branchformer/run.sh b/examples/aishell/branchformer/run.sh index 5b9f9ef0e..f7dda1c9c 100755 --- a/examples/aishell/branchformer/run.sh +++ b/examples/aishell/branchformer/run.sh @@ -15,7 +15,7 @@ stop_stage=5 nj=32 inference_device="cuda" #"cpu" -inference_checkpoint="model.pt.avg5" +inference_checkpoint="model.pt.avg10" inference_scp="wav.scp" inference_batch_size=32 diff --git a/examples/aishell/conformer/run.sh b/examples/aishell/conformer/run.sh index fa6873a06..ba8b43c60 100755 --- a/examples/aishell/conformer/run.sh +++ b/examples/aishell/conformer/run.sh @@ -15,7 +15,7 @@ stop_stage=5 nj=32 inference_device="cuda" #"cpu", "cuda:0", "cuda:1" -inference_checkpoint="model.pt.avg5" +inference_checkpoint="model.pt.avg10" inference_scp="wav.scp" inference_batch_size=1 diff --git a/examples/aishell/e_branchformer/run.sh b/examples/aishell/e_branchformer/run.sh index 49d5af013..bc78b5f01 100755 --- a/examples/aishell/e_branchformer/run.sh +++ b/examples/aishell/e_branchformer/run.sh @@ -15,7 +15,7 @@ stop_stage=5 nj=32 inference_device="cuda" #"cpu" -inference_checkpoint="model.pt.avg5" +inference_checkpoint="model.pt.avg10" inference_scp="wav.scp" inference_batch_size=32 diff --git a/examples/aishell/paraformer/run.sh b/examples/aishell/paraformer/run.sh index 528f1400a..a957b937b 100755 --- a/examples/aishell/paraformer/run.sh +++ b/examples/aishell/paraformer/run.sh @@ -15,7 +15,7 @@ stop_stage=5 nj=32 inference_device="cuda" #"cpu" -inference_checkpoint="model.pt.avg5" +inference_checkpoint="model.pt.avg10" inference_scp="wav.scp" inference_batch_size=32 diff --git a/examples/aishell/transformer/run.sh b/examples/aishell/transformer/run.sh index 336c558db..a5ff7ffcc 100755 --- a/examples/aishell/transformer/run.sh +++ b/examples/aishell/transformer/run.sh @@ -15,7 +15,7 @@ stop_stage=5 nj=32 inference_device="cuda" #"cpu" -inference_checkpoint="model.pt.avg5" +inference_checkpoint="model.pt.avg10" inference_scp="wav.scp" inference_batch_size=32