fix fc neuron count

2024-05-18 18:45:42 +01:00
parent 00749b0374
commit a61e12c5db
1 changed files with 1 additions and 1 deletions
--- 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)