fix(trainers): Hotfix to avoid StopIteration when evaluating with DsTrainer.

This commit is contained in:
Gustavo Rosa 2023-04-06 05:01:25 -07:00
Родитель a19b255b0a
Коммит fed46daa82
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -367,7 +367,7 @@ class DsTrainer(TrainerBase):
"""
eval_dataloader = self._get_dataloader(eval_dataset, shuffle=False)
eval_iterator = iter(eval_dataloader)
eval_iterator = iter(RepeatingLoader(eval_dataloader))
n_eval_steps = self.args.eval_max_steps or len(eval_dataloader)
eval_loss, eval_time = 0.0, time.time()