Upload modified model with logging

config.json

@@ -5,13 +5,13 @@
   "activation_function": "gelu",
   "add_final_layer_norm": false,
   "architectures": [
-    "ModifiedBartForSequenceClassificationWithHook"
+    "BartForSequenceClassification"
   ],
   "attention_dropout": 0.0,
   "auto_map": {
-    "AutoModel": "modeling_bart.ModifiedBartForSequenceClassificationWithHook",
-    "AutoModelForSequenceClassification": "modeling_bart.ModifiedBartForSequenceClassificationWithHook",
-    "AutoConfig": "config.BartConfig"
+    "AutoConfig": "configuring_modified.BartConfig",
+    "AutoForSequenceClassification": "modeling_modified.ModifiedAutoModelForSequenceClassificationWithHook",
+    "AutoModel": "modeling_modified.ModifiedAutoModelWithHook"
   },
   "bos_token_id": 0,
   "classif_dropout": 0.0,

configuring_modified.py

@@ -0,0 +1 @@
+from transformers import BartConfig

modeling_modified.py

@@ -0,0 +1,31 @@
+import logging
+
+import logging
+from transformers import AutoModelForSequenceClassification
+
+logger = logging.getLogger("ModelLogger")
+
+class ModifiedAutoModelForSequenceClassificationWithHook(AutoModelForSequenceClassification):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.register_forward_hook(self.forward_hook)
+
+    @staticmethod
+    def forward_hook(module, inputs, outputs):
+        logger.info(f"Called forward method of {module.__class__.__name__}")
+
+
+import logging
+from transformers import AutoModel
+
+logger = logging.getLogger("ModelLogger")
+
+class ModifiedAutoModelWithHook(AutoModel):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.register_forward_hook(self.forward_hook)
+
+    @staticmethod
+    def forward_hook(module, inputs, outputs):
+        logger.info(f"Called forward method of {module.__class__.__name__}")
+