Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ class MultiModalClassifier(nn.Module):
|
|
| 27 |
self.fc1 = nn.Linear(num_features + self.age_dim + self.anatom_site_dim + self.sex_dim, 256)
|
| 28 |
self.fc2 = nn.Linear(256, num_classes)
|
| 29 |
|
| 30 |
-
# Dropout layer
|
| 31 |
self.dropout = nn.Dropout(p=0.5)
|
| 32 |
|
| 33 |
def forward(self, image, age, anatom_site, sex):
|
|
|
|
| 27 |
self.fc1 = nn.Linear(num_features + self.age_dim + self.anatom_site_dim + self.sex_dim, 256)
|
| 28 |
self.fc2 = nn.Linear(256, num_classes)
|
| 29 |
|
| 30 |
+
# Dropout layer.
|
| 31 |
self.dropout = nn.Dropout(p=0.5)
|
| 32 |
|
| 33 |
def forward(self, image, age, anatom_site, sex):
|