Upload model.py with huggingface_hub
Browse files
model.py
CHANGED
|
@@ -9,7 +9,7 @@ class SimpleCNN(nn.Module):
|
|
| 9 |
nn.Conv2d(32, 64, 3, padding=1), nn.ReLU(), nn.MaxPool2d(2),
|
| 10 |
nn.Conv2d(64, 128, 3, padding=1), nn.ReLU(), nn.AdaptiveAvgPool2d(1),
|
| 11 |
)
|
| 12 |
-
self.fc = nn.Linear(128, 2) #
|
| 13 |
|
| 14 |
def forward(self, x):
|
| 15 |
x = self.net(x)
|
|
|
|
| 9 |
nn.Conv2d(32, 64, 3, padding=1), nn.ReLU(), nn.MaxPool2d(2),
|
| 10 |
nn.Conv2d(64, 128, 3, padding=1), nn.ReLU(), nn.AdaptiveAvgPool2d(1),
|
| 11 |
)
|
| 12 |
+
self.fc = nn.Linear(128, 2) # Negative=0, Positive=1
|
| 13 |
|
| 14 |
def forward(self, x):
|
| 15 |
x = self.net(x)
|