Commit Graph

5 Commits (100830e5efc06f259b347d55edda90f3e75c14f1)

Author SHA1 Message Date
liaoxingyu 6300bd756e Bugfix for cls_layer
In `any_softmax`, all operations are in-place, so pass into the `logits.clone()` to prevent outside logits changed.
2021-05-31 17:32:24 +08:00
Xingyu Liao 1dce15efad
faster dataloader with pre-fetch and cuda stream (#456)
Summary: add a background thread to create a generator with pre-fetch, and create a new cuda stream to copy tensor from cpu to gpu in parallel.

Reviewed by: l1aoxingyu
2021-04-12 15:03:35 +08:00
liaoxingyu 55300730e1 update fastreid v1.2 readme and changelog 2021-04-06 20:09:13 +08:00
liaoxingyu 1666c82db4 update readme about FastRT 2021-01-30 16:11:08 +08:00
liaoxingyu 77a91b1204 feat: support multi-teacher kd
Summary: support multi-teacher kd with logits and overhaul distillation
2021-01-29 17:25:31 +08:00