From 69072f0d4ccb13d7c089cfcc8b47d2d5d3354800 Mon Sep 17 00:00:00 2001 From: Mddct Date: Fri, 22 Mar 2024 17:44:20 +0800 Subject: [PATCH 1/2] fix modify --- examples/aishell/paraformer/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/aishell/paraformer/README.md b/examples/aishell/paraformer/README.md index 64482c44d..0b67acf7b 100644 --- a/examples/aishell/paraformer/README.md +++ b/examples/aishell/paraformer/README.md @@ -7,7 +7,7 @@ mkdir -p ${output_dir} --output_dir ${output_dir} # init ctc and embed(used in sampler) python local/modify_ckpt.py \ - --add_list "{\"ctc.ctc_lo.weight\": \"decoder.embed.0.weight\", \"embed.0.weight\": \"decoder.embed.0.weight\"}" \ + --add_list "{\"ctc.ctc_lo.weight\": \"embed.0.weight\"}" \ --input_ckpt exp/paraformer/large/wenet_paraformer.pt \ --output_ckpt exp/paraformer/large/wenet_paraformer.init-ctc.init-embed.pt ``` From 68a0109544ace3ec59689217aa75baf6b8c61846 Mon Sep 17 00:00:00 2001 From: Dinghao Zhou Date: Fri, 22 Mar 2024 17:48:50 +0800 Subject: [PATCH 2/2] Update README.md --- examples/aishell/paraformer/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/aishell/paraformer/README.md b/examples/aishell/paraformer/README.md index 0b67acf7b..9fe648842 100644 --- a/examples/aishell/paraformer/README.md +++ b/examples/aishell/paraformer/README.md @@ -7,7 +7,7 @@ mkdir -p ${output_dir} --output_dir ${output_dir} # init ctc and embed(used in sampler) python local/modify_ckpt.py \ - --add_list "{\"ctc.ctc_lo.weight\": \"embed.0.weight\"}" \ + --add_list "{\"ctc.ctc_lo.weight\": \"embed.weight\"}" \ --input_ckpt exp/paraformer/large/wenet_paraformer.pt \ --output_ckpt exp/paraformer/large/wenet_paraformer.init-ctc.init-embed.pt ```