fix dpo eval override to call grandparent instead of the broken super (#2628) [skip ci]
This commit is contained in:
@@ -247,7 +247,9 @@ class AxolotlDPOTrainer(RngLoaderMixin, SchedulerMixin, DPOTrainer):
|
|||||||
)
|
)
|
||||||
|
|
||||||
# Base evaluation
|
# Base evaluation
|
||||||
initial_output = super().evaluation_loop(
|
initial_output = super( # pylint: disable=bad-super-call
|
||||||
|
DPOTrainer, self
|
||||||
|
).evaluation_loop(
|
||||||
dataloader,
|
dataloader,
|
||||||
description,
|
description,
|
||||||
prediction_loss_only,
|
prediction_loss_only,
|
||||||
|
|||||||
Reference in New Issue
Block a user