Spaces:
Runtime error
Runtime error
Update app/Hackathon_setup/exp_recognition_model.py
Browse files
app/Hackathon_setup/exp_recognition_model.py
CHANGED
|
@@ -37,6 +37,7 @@ class facExpRec(torch.nn.Module):
|
|
| 37 |
def fclayer(self, in_features, out_features):
|
| 38 |
return nn.Sequential(
|
| 39 |
nn.Linear(in_features, out_features),
|
|
|
|
| 40 |
nn.Dropout1d(0.4),
|
| 41 |
nn.ReLU(),
|
| 42 |
)
|
|
|
|
| 37 |
def fclayer(self, in_features, out_features):
|
| 38 |
return nn.Sequential(
|
| 39 |
nn.Linear(in_features, out_features),
|
| 40 |
+
nn.BatchNorm1d(out_features),
|
| 41 |
nn.Dropout1d(0.4),
|
| 42 |
nn.ReLU(),
|
| 43 |
)
|