Commit Graph

4 Commits (91ff63118434fc427dbf4fd693b16b9e8a4b436d)

Author SHA1 Message Date
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