Increase map embeddings size

thatguy11325 · Aug 30, 2024 · 8e2f8bf · 8e2f8bf
1 parent e9e0c15
commit 8e2f8bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pokemonred_puffer/policies/multi_convolutional.py b/pokemonred_puffer/policies/multi_convolutional.py
@@ -96,7 +96,7 @@ def __init__(
 
         # pokemon has 0xF7 map ids
         # Lets start with 4 dims for now. Could try 8
-        self.map_embeddings = nn.Embedding(0xF7, 4, dtype=torch.float32)
+        self.map_embeddings = nn.Embedding(0xFF, 4, dtype=torch.float32)
         # N.B. This is an overestimate
         item_count = max(Items._value2member_map_.keys())
         self.item_embeddings = nn.Embedding(