From 0486bc37a326979b7fe4bb3f505e7b3a6babe662 Mon Sep 17 00:00:00 2001 From: topduke <784990967@qq.com> Date: Wed, 18 Aug 2021 19:53:52 +0800 Subject: [PATCH] modify beam size --- configs/rec/rec_mtb_nrtr.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/rec/rec_mtb_nrtr.yml b/configs/rec/rec_mtb_nrtr.yml index c832b9170..171ac7e3f 100644 --- a/configs/rec/rec_mtb_nrtr.yml +++ b/configs/rec/rec_mtb_nrtr.yml @@ -46,7 +46,7 @@ Architecture: name: TransformerOptim d_model: 512 num_encoder_layers: 6 - beam_size: -1 # When Beam size is greater than 0, it means to use beam search when evaluation. + beam_size: 10 # When Beam size is greater than 0, it means to use beam search when evaluation. Loss: