From a61e12c5dba7fe37549ade9259dd04ca3549dfd2 Mon Sep 17 00:00:00 2001 From: Kenneth Date: Sat, 18 May 2024 18:45:42 +0100 Subject: [PATCH] fix fc neuron count --- resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/resnet.py b/resnet.py index ce3ddc7..925f058 100644 --- a/resnet.py +++ b/resnet.py @@ -175,7 +175,7 @@ class MaiRes(nn.Module): ) self.avgpool = nn.AvgPool2d(kernel_size=RESNET_KERNEL_SIZE) - self.fc = nn.Linear(in_features=1000, out_features=1) + self.fc = nn.Linear(in_features=8192, out_features=1) def forward(self, x): x = self.conv(x)