Skip to content

Commit

Permalink
Removed redundant metrics argument.
Browse files Browse the repository at this point in the history
  • Loading branch information
seanswyi committed Jul 6, 2024
1 parent 1e16bd2 commit a114602
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -2805,7 +2805,7 @@ def _maybe_log_save_evaluate(self, tr_loss, grad_norm, model, trial, epoch, igno
self.control_should_save = False

if self.control.should_save:
self._save_checkpoint(model, trial, metrics=metrics)
self._save_checkpoint(model, trial)
self.control = self.callback_handler.on_save(self.args, self.state, self.control)

def _load_rng_state(self, checkpoint):
Expand Down

0 comments on commit a114602

Please sign in to comment.