Accelerator .prepare() replaces custom DataLoader Sampler

[muellerzr]

@HTess

Is this a normal behaviour? Since it seems I can’t manually restore my sampler afterwhile, this is quite a problem.

Yes. The original sampler is still there I think as well? Can you check sampler.sampler? Or sampler.batch_sampler