Thompson001 commited on
Commit
a667a5b
·
verified ·
1 Parent(s): c12e0d7

Upload model.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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) # crack / no_crack
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)