This commit is contained in:
Eren Golge committed 2018-03-23 05:18:51 -07:00
1 parent 2617518d91
commit df4a644326
1 file changed
+1 -1
+1 -1
View File
@@ -253,7 +253,7 @@ class Decoder(nn.Module):
B = inputs.size(0)
# Run greedy decoding if memory is None
greedy = ~self.training
greedy = not self.training
if memory is not None:
# Grouping multiple frames if necessary