Bring back `set_epoch` for Accelerate-based dataloaders (#26850)
* Working tests!
* Fix sampler
* Fix
* Update src/transformers/trainer.py
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
* Fix check
* Clean
---------
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>