quick sanity

Parry-Parry · Nov 6, 2024 · 2db4c7b · 2db4c7b
1 parent c2bdcdb
commit 2db4c7b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)