moazx commited on
Commit
02b7bab
·
verified ·
1 Parent(s): 832f210

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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):