p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!11051 Fixed GNMT reasoning performance 

From: @gaojing22
Reviewed-by: @wuxuejian,@linqingke
Signed-off-by: @wuxuejian
This commit is contained in:
mindspore-ci-bot 2021-01-08 10:36:37 +08:00 committed by Gitee
parent d6bf058052 2c4f7d92b7
commit 64b72bd511
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model_zoo/official/nlp/gnmt_v2/src/gnmt_model/beam_search.py
View File

@ -172,7 +172,7 @@ class BeamSearchDecoder(nn.Cell):
max_decode_length=64,
sos_id=2,
eos_id=3,
is_using_while=False,
is_using_while=True,
compute_type=mstype.float32):
super(BeamSearchDecoder, self).__init__()
@ -408,6 +408,7 @@ class BeamSearchDecoder(nn.Cell):
state_seq, state_length, None, decoder_hidden_state, accu_attn_scores,
state_finished)
else:
# At present, only ascend910 supports while operation.
idx = self.start + 1
ends = self.start + self.max_decode_length + 1
while idx < ends: