change channel count for nfl-nn

guidopetri · Mar 21, 2021 · b00b805 · b00b805
1 parent 37c8cae
commit b00b805
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/csgo_wp/model.py b/csgo_wp/model.py
@@ -458,7 +458,7 @@ def __init__(self,
         # we're just doing the same model as:
         # https://github.com/juancamilocampos/nfl-big-data-bowl-2020/blob/master/1st_place_zoo_solution_v2.ipynb
 
-        self.conv1 = torch.nn.Conv2d(10, 128, kernel_size=1, stride=1)
+        self.conv1 = torch.nn.Conv2d(7, 128, kernel_size=1, stride=1)
         self.conv2 = torch.nn.Conv2d(128, 160, kernel_size=1, stride=1)
         self.conv3 = torch.nn.Conv2d(160, 128, kernel_size=1, stride=1)
 
@@ -585,7 +585,7 @@ def forward(self, x):
 
     print('\nTesting NFL-NN')
 
-    t = torch.rand(size=(5, 10, 5, 5))
+    t = torch.rand(size=(5, 7, 5, 5))
 
     mod = NFL_NN()