change card types

LucasColas · Sep 7, 2023 · 238e791 · 238e791
1 parent 629f575
commit 238e791
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/PokerPlus/DeepCFR/deep_cfr.py b/PokerPlus/DeepCFR/deep_cfr.py
@@ -33,7 +33,7 @@ def deep_cfr(
     nb_game_tree_traversals: int = 200,
     game: TexasHoldEm = None,
     n_actions: int = 3,
-    n_card_types: int = 4,
+    n_card_types: int = 52,
     n_bets: int = 20,
 ):
     """

diff --git a/PokerPlus/DeepCFR/nn.py b/PokerPlus/DeepCFR/nn.py
@@ -63,8 +63,11 @@ def forward(self, cards, bets):
             print("do embedding")
             print("card_group: ", card_group)
             if card_group.numel():
-                card_embs.append(embedding(card_group.view(1, -1)))
+                card_embs.append(embedding(card_group.view(-1, 1)))
+
         card_embs = torch.cat(card_embs, dim=1)
+
+        print("cards embs shape : ", card_embs.shape)
         x = F.relu(self.card1(card_embs))
         x = F.relu(self.card2(x))
         x = F.relu(self.card3(x))