increase hidden layer size

DLR-MF-DAS · Jan 17, 2025 · fd18c31 · fd18c31
1 parent a6eba26
commit fd18c31
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/methane_super_emitters/model.py b/src/methane_super_emitters/model.py
@@ -6,13 +6,13 @@
 class SuperEmitterDetector(L.LightningModule):
     def __init__(self):
         super().__init__()
-        self.cnv = nn.Conv2d(1, 32, 5, 4)
+        self.cnv = nn.Conv2d(1, 64, 5, 4)
         self.rel = nn.ReLU()
-        self.bn = nn.BatchNorm2d(32)
+        self.bn = nn.BatchNorm2d(64)
         self.mxpool = nn.MaxPool2d(4)
         self.flat = nn.Flatten()
-        self.fc1 = nn.Linear(32, 16)
-        self.fc2 = nn.Linear(16, 1)
+        self.fc1 = nn.Linear(64, 32)
+        self.fc2 = nn.Linear(32, 1)
         self.sigmoid = nn.Sigmoid()
         self.accuracy = torchmetrics.classification.BinaryAccuracy()