!2113 Modify SentimentNet to avoid performance decay

Merge pull request !2113 from chenweifeng/sentiment
2020-06-15 14:15:27 +08:00 · 2020-06-15 14:15:27 +08:00 · 5036222880
parent c230cfdd09 deffd6a638
commit 5036222880
1 changed files with 1 additions and 1 deletions
--- a/mindspore/model_zoo/lstm.py
+++ b/mindspore/model_zoo/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