From 2db4c7bcf862536bbb4dd9f1bbd773dbf4d5116d Mon Sep 17 00:00:00 2001
From: Andrew Parry <a.parry.1@research.gla.ac.uk>
Date: Wed, 6 Nov 2024 16:56:24 +0000
Subject: [PATCH] quick sanity

---
 rankers/train/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/rankers/train/trainer.py b/rankers/train/trainer.py
index 58e0c14..2dadd4c 100644
--- a/rankers/train/trainer.py
+++ b/rankers/train/trainer.py
@@ -27,6 +27,7 @@ def __init__(self, loss_fn=None, **kwargs) -> None:
             self.loss = loss_fn
         self.tokenizer = self.data_collator.tokenizer
         self.model.config.group_size = self.args.group_size
+        print(self.callback_handler.callbacks)
 
     def compute_loss(self, model, inputs, return_outputs=False):
         outputs = model(self.loss, **inputs)