athenlab/mai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix fc neuron count

Browse Source
This commit is contained in:
Kenneth
2024-05-18 18:45:42 +01:00
parent 00749b0374
commit a61e12c5db
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
resnet.py
View File

@@ -175,7 +175,7 @@ class MaiRes(nn.Module):
) )
self.avgpool = nn.AvgPool2d(kernel_size=RESNET_KERNEL_SIZE) 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): def forward(self, x):
x = self.conv(x) x = self.conv(x)