modify beam size

This commit is contained in:
topduke
2021-08-18 19:53:52 +08:00
committed by GitHub
parent adc62fcde4
commit 0486bc37a3
+1 -1
View File
@@ -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: