diff --git a/examples/aishell/branchformer/run.sh b/examples/aishell/branchformer/run.sh index ecdfd24c9..5b9f9ef0e 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" +inference_checkpoint="model.pt.avg5" inference_scp="wav.scp" inference_batch_size=32 diff --git a/examples/aishell/conformer/run.sh b/examples/aishell/conformer/run.sh index 19aa2db6a..fa6873a06 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" +inference_checkpoint="model.pt.avg5" 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 e5a8569ff..49d5af013 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" +inference_checkpoint="model.pt.avg5" inference_scp="wav.scp" inference_batch_size=32 diff --git a/examples/aishell/paraformer/run.sh b/examples/aishell/paraformer/run.sh index 3c7feb2a1..a602e283d 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" +inference_checkpoint="model.pt.avg5" inference_scp="wav.scp" inference_batch_size=32 diff --git a/examples/aishell/transformer/run.sh b/examples/aishell/transformer/run.sh index e9c3d6414..336c558db 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" +inference_checkpoint="model.pt.avg5" inference_scp="wav.scp" inference_batch_size=32