mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-03 20:22:49 +08:00
fix ppo save model
Former-commit-id: 7ba57d5b1469cd0de0bb391b915bedec97b20ebd
This commit is contained in:
parent
4e86462bad
commit
c8780205bc
@ -147,7 +147,8 @@ class CustomPPOTrainer(PPOTrainer, Trainer):
|
||||
dataiter = iter(self.dataloader)
|
||||
steps_trained = 0
|
||||
|
||||
self.log_callback.on_train_end(
|
||||
self.log_callback.on_train_end(self.args, self.state, self.control)
|
||||
self.save_callback.on_train_end(
|
||||
self.args, self.state, self.control, model=self.accelerator.unwrap_model(self.model)
|
||||
)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user