Maxlegrec
/

ChessBot

Feature Extraction

Model card Files Files and versions Community

Maxlegrec commited on Jul 11

Commit

ba34655

·

verified ·

1 Parent(s): a24833c

Update modeling_chessbot.py

Files changed (1) hide show

modeling_chessbot.py +1 -1

modeling_chessbot.py CHANGED Viewed

@@ -613,7 +613,7 @@ class ChessBotModel(ChessBotPreTrainedModel):
                 # Find the move with the highest policy value that is legal
                 legal_moves_mask = - torch.ones_like(policy) * 999
                 for move in legal_moves:
-                        legal_moves_mask[policy_index[move]] = 0
                 policy = legal_moves_mask + policy
                 return policy_index[torch.argmax(policy).item()]
             else:

                 # Find the move with the highest policy value that is legal
                 legal_moves_mask = - torch.ones_like(policy) * 999
                 for move in legal_moves:
+                        legal_moves_mask[policy_index.index(move)] = 0
                 policy = legal_moves_mask + policy
                 return policy_index[torch.argmax(policy).item()]
             else: