diff --git a/model_zoo/lstm/src/lstm.py b/model_zoo/lstm/src/lstm.py index f014eef8df0..c3ca0bbf7c9 100644 --- a/model_zoo/lstm/src/lstm.py +++ b/model_zoo/lstm/src/lstm.py @@ -88,6 +88,6 @@ class SentimentNet(nn.Cell): embeddings = self.trans(embeddings, self.perm) output, _ = self.encoder(embeddings, (self.h, self.c)) # states[i] size(64,200) -> encoding.size(64,400) - encoding = self.concat((output[0], output[-1])) + encoding = self.concat((output[0], output[499])) outputs = self.decoder(encoding) return outputs