fix bug in discrete_net.py (#10)

2020-03-31 16:13:53 +08:00 · 2020-03-31 16:13:53 +08:00 · 98feb79057
commit 98feb79057
parent 04208e6cce
1 changed files with 1 additions and 1 deletions
--- a/examples/discrete_net.py
+++ b/examples/discrete_net.py
@ -34,7 +34,7 @@ class Actor(nn.Module):

    def forward(self, s, state=None, info={}):
        logits, h = self.preprocess(s, state)
-        logits = F.softmax(logits, dim=-1)
+        logits = F.softmax(self.last(logits), dim=-1)
        return logits, h