Update model/modeling_modified.py

model/modeling_modified.py

@@ -1,5 +1,3 @@
-import logging
-
 import logging
 from transformers import BartForSequenceClassification
 
@@ -13,6 +11,7 @@ class ModifiedBartForSequenceClassificationWithHook(BartForSequenceClassificatio
     @staticmethod
     def forward_hook(module, inputs, outputs):
         logger.info(f"Called forward method of {module.__class__.__name__}")
+        print(f"Called forward method of {module.__class__.__name__}")
 
 
 import logging
@@ -28,4 +27,5 @@ class ModifiedAutoModelWithHook(AutoModel):
     @staticmethod
     def forward_hook(module, inputs, outputs):
         logger.info(f"Called forward method of {module.__class__.__name__}")
+        print(f"Called forward method of {module.__class__.__name__}")