We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a5ef2c4 commit 2bd8785Copy full SHA for 2bd8785
neuralmonkey/decoders/autoregressive.py
@@ -450,7 +450,7 @@ def state_to_logits(state: tf.Tensor) -> tf.Tensor:
450
451
if self.supress_unk:
452
unk_mask = tf.one_hot(
453
- UNK_TOKEN_INDEX, depth=len(self.vocabulary), on_value=-1e0)
+ UNK_TOKEN_INDEX, depth=len(self.vocabulary), on_value=-1e9)
454
logits += unk_mask
455
456
return logits
0 commit comments