Update app/Hackathon_setup/exp_recognition_model.py

app/Hackathon_setup/exp_recognition_model.py

@@ -2,6 +2,9 @@ import torch
 import torchvision
 import torch.nn as nn
 from torchvision import transforms
+import transformers
+
+transformers.logging.set_verbosity_info()
 ## Add more imports if required
 
 ####################################################################################################################
@@ -45,13 +48,14 @@ class facExpRec(torch.nn.Module):
 
     
     def forward(self, x):
+        logger.info(x.shape)
         x = self.conv1(x)
         x = self.conv2(x)
         x = self.conv3(x)
         x = self.conv4(x)
-        print(x.shape)
+        logger.info(x.shape)
         x = x.view(-1, 512)
-        print(x.shape)
+        logger.info(x.shape)
         x = self.fc1(x)
         x = self.fc2(x)
         x = self.fc3(x)