utilize LogSoftmax.exp()

masus04 · Sep 14, 2018 · 98f57f0 · 98f57f0
1 parent 12d5ef9
commit 98f57f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/abstractClasses.py b/abstractClasses.py
@@ -330,7 +330,7 @@ def forward(self, input, legal_moves_map):
             x = input
             # set illegal moves to zero, softmax, set illegal moves to zero again
             # x = input * legal_moves_map
-            x = self.softmax(x)
+            x = self.softmax(x).exp()
             x = x * legal_moves_map
 
             return x