We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bb017a8 commit 73a1b44Copy full SHA for 73a1b44
swift/trainers/kto_trainer.py
@@ -87,7 +87,7 @@ def __init__(self, *args, template: Template, test_oom_error=False, **kwargs):
87
super().__init__(*args, **kwargs)
88
train_ds_info = self.stat_dataset(self.train_dataset)
89
if self.eval_dataset is not None:
90
- val_ds_info = self.stat_dataset(self.eval_dataset, self.is_encoder_decoder)
+ val_ds_info = self.stat_dataset(self.eval_dataset)
91
self.dataset_info = {'train_dataset': train_ds_info, 'val_dataset': val_ds_info}
92
else:
93
self.dataset_info = {'train_dataset': train_ds_info}
0 commit comments