I want to use a DataLoader that uses a custom sampler you can find at vision/references/classification/sampler.py at main · pytorch/vision · GitHub When doing : print(dataset, dataset.sampler) dataset = accelerator.prepare(dataset) print(dataset, dataset.sampler) I get the following print: <torc…

Accelerator .prepare() replaces custom DataLoader Sampler

razvan-a-matisan March 9, 2025, 3:03pm 6

Hi! Is there a way to get the state of the custom sampler after .prepare() method was used?
I would like to save the state of the custom sampler, but I am not sure if it’s possible?

1 Like

Topic		Replies	Views
Accelerate - WeightedRandomSampler Dataloader Intermediate	1	247	June 18, 2024
Can accelerator handle the distributed sampler? 🤗Accelerate	2	2903	December 21, 2021
Using DistributedSampler with accelerate 🤗Transformers	4	117	April 2, 2025
Troubles with features in .prepare() 🤗Accelerate	1	34	November 30, 2024
DataLoader from accelerator samples from beginning of dataset for last batch 🤗Accelerate	1	648	January 15, 2024

Accelerator .prepare() replaces custom DataLoader Sampler

Related topics