parameterized custom sampler to allow weighted sampling

utterworks / fast-bert

Super easy library for BERT based NLP models

Apache License 2.0

1.85k stars 342 forks source link

parameterized custom sampler to allow weighted sampling #218

Closed aaronbriel closed 4 years ago

aaronbriel commented 4 years ago

Allows one to pass a custom sampler to the training DataLoader. For example, one can now pass a WeightedRandomSampler to implement oversampling on underrepresented classes.

aaronbriel commented 4 years ago

@kaushaltrivedi any way to merge this?

kaushaltrivedi commented 4 years ago

Merged. Sorry for the delay