From 15b621e3596a4ff642fde01226f3d5a0d9a05b6c Mon Sep 17 00:00:00 2001 From: "xiangyu.xie" Date: Mon, 27 Oct 2025 18:19:17 +0100 Subject: [PATCH] update singlePhotonNet_251022 --- src/models.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/models.py b/src/models.py index 86bab55..9086b25 100644 --- a/src/models.py +++ b/src/models.py @@ -81,17 +81,17 @@ class singlePhotonNet_251022(nn.Module): def __init__(self): 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.conv3 = nn.Conv2d(10, 20, kernel_size=3, padding=1) - self.gap = nn.AdaptiveAvgPool2d(1) - self.fc = nn.Linear(20, 2) - + self.conv3 = nn.Conv2d(10, 20, kernel_size=3) + self.fc = nn.Linear(20, 3) + self.weight_init() + def forward(self, x): x = F.relu(self.conv1(x)) x = F.relu(self.conv2(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) return x