From 97dd199b382fd0466e48a3cf7a87bc010550897e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Mon, 25 Mar 2024 23:01:19 +0800 Subject: [PATCH] train --- examples/aishell/branchformer/run.sh | 2 +- examples/aishell/conformer/run.sh | 2 +- examples/aishell/e_branchformer/run.sh | 2 +- examples/aishell/paraformer/run.sh | 2 +- examples/aishell/transformer/run.sh | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) 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