Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,6 @@ class SimpleCNN(nn.Module):
|
|
| 11 |
self.pool = nn.MaxPool2d(kernel_size=2, stride=2)
|
| 12 |
self.relu = nn.ReLU()
|
| 13 |
self.dropout = nn.Dropout(0.5)
|
| 14 |
-
# initialize fc layers dynamically based on a dummy input
|
| 15 |
self._initialize_fc(num_classes)
|
| 16 |
|
| 17 |
def _initialize_fc(self, num_classes):
|
|
|
|
| 11 |
self.pool = nn.MaxPool2d(kernel_size=2, stride=2)
|
| 12 |
self.relu = nn.ReLU()
|
| 13 |
self.dropout = nn.Dropout(0.5)
|
|
|
|
| 14 |
self._initialize_fc(num_classes)
|
| 15 |
|
| 16 |
def _initialize_fc(self, num_classes):
|