GPU Lstm network

fd2fa2bc · wilfChen · cf4d317d · fd2fa2bc
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

model_zoo/lstm/src/lstm.py model_zoo/lstm/src/lstm.py +1 -1

未找到文件。
--- 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