how to apply pytorch's WeightedRandomSampler

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)

https://ffcv.io

Apache License 2.0

2.84k stars 178 forks source link

Open Nise-2-meet-U opened 2 years ago

Nise-2-meet-U commented 2 years ago

@GuillaumeLeclerc

arunpatro-meesho commented 2 years ago

wouterzwerink commented 1 year ago

https://github.com/libffcv/ffcv/issues/301#issuecomment-1521627945 Heres my workaround to use WeightedRandomSampler in an ffcv Loader