From b00b805c3ab7780d88bc3f5083bdcb735648f152 Mon Sep 17 00:00:00 2001 From: Guido Petri <18634426+guidopetri@users.noreply.github.com> Date: Sun, 21 Mar 2021 19:24:16 -0400 Subject: [PATCH] change channel count for nfl-nn --- csgo_wp/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/csgo_wp/model.py b/csgo_wp/model.py index b34f830..7a395df 100755 --- 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()