!2777 gpu lstm network modify

Merge pull request !2777 from chenweifeng/lstm_performace
pull/2777/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 89e9f0d4bb

@ -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

Loading…
Cancel
Save