fix resume from epoch >= 1

Author: ngazagna-qc

src/transformers/trainer.py

@@ -2414,6 +2414,9 @@ def _inner_training_loop(
             if epoch == epochs_trained and resume_from_checkpoint is not None:
                 if steps_trained_in_current_epoch > 0 and not args.ignore_data_skip:
                     epoch_dataloader = skip_first_batches(epoch_dataloader, steps_trained_in_current_epoch)
+                    if hasattr(epoch_dataloader, "set_epoch"):
+                        epoch_dataloader.set_epoch(epoch)
+
                     step = steps_trained_in_current_epoch - 1
                     rng_to_sync = True
                 elif steps_trained_in_current_epoch == 0: