How to customize dataloader creation in trainer?

No, you should probably use a manual training loop for that, using the Accelerate library.