Update app/Hackathon_setup/exp_recognition_model.py

app/Hackathon_setup/exp_recognition_model.py

@@ -2,6 +2,11 @@ import torch
 import torchvision
 import torch.nn as nn
 from torchvision import transforms
+from transformers.utils import logging
+
+logging.set_verbosity_info()
+logger = logging.get_logger("transformers")
+
 ## Add more imports if required
 
 ####################################################################################################################
@@ -45,12 +50,20 @@ class facExpRec(torch.nn.Module):
     
     def forward(self, x):
         x = self.conv1(x)
+        logger.info(x.shape)
         x = self.conv2(x)
+        logger.info(x.shape)
         x = self.conv3(x)
+        logger.info(x.shape)
+        logger.info(x.shape)
         x = x.view(-1, 512*3*3)
+        logger.info(x.shape)
         x = self.fc1(x)
+        logger.info(x.shape)
         x = self.fc2(x)
+        logger.info(x.shape)
         x = self.fc3(x)
+        logger.info(x.shape)
         return x
         
 # Sample Helper function