hmm

Parry-Parry · Nov 6, 2024 · 1e0d14e · 1e0d14e
1 parent b622499
commit 1e0d14e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rankers/train/training_arguments.py b/rankers/train/training_arguments.py
@@ -34,7 +34,7 @@ class RankerTrainingArguments(TrainingArguments):
         default=None,
         metadata={"help": "Wandb project name"}
     )
-    loss_fn : Optional[Union[str, callable]] = field(
+    loss_fn : Optional[str] = field(
         default='lce',
         metadata={"help": "Loss function to use"}
     )