search

eora-ai / torchok

Production-oriented Computer Vision models training pipeline for common tasks: classification, segmentation, detection and representationšŸ„¤
https://torchok.readthedocs.org
Apache License 2.0
54 stars 6 forks source link

Add weighted samplers #189

Closed kristinarakova closed 1 year ago

kristinarakova commented 1 year ago

Added WeightedRandomSampler from PyTorch