Update LM.py

2019-09-17 20:50:02 +08:00 · 2019-09-17 20:50:02 +08:00 · e57fe43915
parent 36b2ecb8a2
commit e57fe43915
1 changed files with 1 additions and 1 deletions
--- a/deepke/model/LM.py
+++ b/deepke/model/LM.py
@ -15,6 +15,6 @@ class LM(BasicModule):

    def forward(self, x):
        x = x[0]
-        out = self.lm(x)[-1]
+        out = self.lm(x)[0][:,0]
        out = self.fc(out)
        return out