Commit Graph

3 Commits (c9bc3ceb2f7a6438b62fb515ea3df6d1e999e95d)

Author SHA1 Message Date
liaoxingyu de81b3dbaa fix ClasEvalution too much memory cost
Compute total number of correct predictions on each batch avoiding keeping all predicted logits, which will cost too much memory when the number of classes is large

#503
2021-06-07 15:48:47 +08:00
liaoxingyu 0c8e3d9805 update imbalanced sampler
Summary: add a new sampler, which is useful for imbalanced or long-tail dataset. This refers to ufoym/imbalanced-dataset-sampler.
2021-04-21 17:05:10 +08:00
Xingyu Liao 890224f25c
support classification in fastreid (#443)
Summary: support classification and refactor build_dataloader which can support explicit parameters passing
2021-03-26 20:17:39 +08:00