Spaces:

tsaddev
/

facerecognization

Runtime error

tsaddev commited on Oct 29, 2023

Commit

594a67c

1 Parent(s): 92b27be

Update app/Hackathon_setup/exp_recognition_model.py

Files changed (1) hide show

app/Hackathon_setup/exp_recognition_model.py CHANGED Viewed

@@ -37,7 +37,6 @@ class facExpRec(torch.nn.Module):
     def fclayer(self, in_features, out_features):
         return nn.Sequential(
             nn.Linear(in_features, out_features),
-            nn.BatchNorm1d(out_features),
             nn.Dropout1d(0.4),
             nn.ReLU(),
         )

     def fclayer(self, in_features, out_features):
         return nn.Sequential(
             nn.Linear(in_features, out_features),
             nn.Dropout1d(0.4),
             nn.ReLU(),
         )