423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2026-04-24 16:48:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

[v1] fix epoch and steps (#10422)

Browse Source
This commit is contained in:
jiaqiw09
2026-04-23 17:29:06 +08:00
committed by GitHub
parent ecca167eb4
commit e0bc3c1971
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/llamafactory/v1/core/base_trainer.py
View File

@@ -227,7 +227,9 @@ class BaseTrainer:
"""Train the model."""
self.model.train()
self.callback_handler.on_train_begin(self.args, self.state)
for epoch in range(self._resume_epoch, self.args.num_train_epochs):
epoch = self._resume_epoch
while self.global_step < self.num_training_steps:
self.state.epoch = epoch
self.train_batch_generator.set_epoch(epoch)
self.callback_handler.on_epoch_begin(self.args, self.state)
@@ -332,6 +334,7 @@ class BaseTrainer:
return
self.callback_handler.on_epoch_end(self.args, self.state)
epoch += 1
self.callback_handler.on_train_end(self.args, self.state)