check if a tensor if empty

LucasColas · Sep 4, 2023 · 629f575 · 629f575
1 parent 0372930
commit 629f575
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/PokerPlus/DeepCFR/nn.py b/PokerPlus/DeepCFR/nn.py
@@ -62,7 +62,8 @@ def forward(self, cards, bets):
         for embedding, card_group in zip(self.card_embeddings, cards):
             print("do embedding")
             print("card_group: ", card_group)
-            card_embs.append(embedding(card_group.view(1, -1)))
+            if card_group.numel():
+                card_embs.append(embedding(card_group.view(1, -1)))
         card_embs = torch.cat(card_embs, dim=1)
         x = F.relu(self.card1(card_embs))
         x = F.relu(self.card2(x))