Spaces:

tsaddev
/

facerecognization

Runtime error

tsaddev commited on Oct 29, 2023

Commit

1a4a2dc

1 Parent(s): 2562a04

Update app/Hackathon_setup/exp_recognition_model.py

Files changed (1) hide show

app/Hackathon_setup/exp_recognition_model.py CHANGED Viewed

@@ -49,7 +49,9 @@ class facExpRec(torch.nn.Module):
         x = self.conv2(x)
         x = self.conv3(x)
         x = self.conv4(x)
         x = x.view(-1, 512)
         x = self.fc1(x)
         x = self.fc2(x)
         x = self.fc3(x)

         x = self.conv2(x)
         x = self.conv3(x)
         x = self.conv4(x)
+        print(x.shape)
         x = x.view(-1, 512)
+        print(x.shape)
         x = self.fc1(x)
         x = self.fc2(x)
         x = self.fc3(x)