MLXID/DeepLearning
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update singlePhotonNet_251022

Browse Source
This commit is contained in:
xie_x1
2025-10-27 18:19:17 +01:00
parent 019584d2bb
commit 15b621e359
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/models.py
View File

@@ -81,17 +81,17 @@ class singlePhotonNet_251022(nn.Module):
def __init__(self): def __init__(self):
super(singlePhotonNet_251022, self).__init__() super(singlePhotonNet_251022, self).__init__()
self.conv1 = nn.Conv2d(1, 5, kernel_size=3, padding=1) self.conv1 = nn.Conv2d(3, 5, kernel_size=3, padding=1)
self.conv2 = nn.Conv2d(5, 10, kernel_size=3, padding=1) self.conv2 = nn.Conv2d(5, 10, kernel_size=3, padding=1)
self.conv3 = nn.Conv2d(10, 20, kernel_size=3, padding=1) self.conv3 = nn.Conv2d(10, 20, kernel_size=3)
self.gap = nn.AdaptiveAvgPool2d(1) self.fc = nn.Linear(20, 3)
self.fc = nn.Linear(20, 2) self.weight_init()
def forward(self, x): def forward(self, x):
x = F.relu(self.conv1(x)) x = F.relu(self.conv1(x))
x = F.relu(self.conv2(x)) x = F.relu(self.conv2(x))
x = F.relu(self.conv3(x)) x = F.relu(self.conv3(x))
x = self.gap(x).view(x.size(0), -1) x = x.view(x.size(0), -1)
x = self.fc(x) x = self.fc(x)
return x return x