fix loop (#6698)

Meiyim · web-flow · commit ac6f9b7e3aee · 2023-08-12T01:47:45.000+08:00
diff --git a/paddlenlp/trainer/trainer.py b/paddlenlp/trainer/trainer.py
@@ -847,7 +847,7 @@ def train(
                     )
 
                     self.state.global_step += 1
-                    self.state.epoch = epoch + self.state.global_step / steps_in_epoch
+                    self.state.epoch = epoch + (step + 1) / steps_in_epoch
                     self.control = self.callback_handler.on_step_end(args, self.state, self.control)
                     self._maybe_log_save_evaluate(tr_loss, model, epoch, ignore_keys_for_eval, inputs=inputs)
                     self._print_timer()

Original file line number	Diff line number	Diff line change
`@@ -847,7 +847,7 @@ def train(`
`847`	`847`	`)`
`848`	`848`
`849`	`849`	`self.state.global_step += 1`
`850`		`- self.state.epoch = epoch + self.state.global_step / steps_in_epoch`
	`850`	`+ self.state.epoch = epoch + (step + 1) / steps_in_epoch`
`851`	`851`	`self.control = self.callback_handler.on_step_end(args, self.state, self.control)`
`852`	`852`	`self._maybe_log_save_evaluate(tr_loss, model, epoch, ignore_keys_for_eval, inputs=inputs)`
`853`	`853`	`self._print_timer()`