fast-reid/fastreid/modeling
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
..
backbones support repvgg (#429) 2021-03-11 14:23:24 +08:00
heads update fastreid v1.2 2021-04-02 21:33:13 +08:00
losses update docs 2021-01-23 15:25:58 +08:00
meta_arch faster dataloader with pre-fetch and cuda stream (#456) 2021-04-12 15:03:35 +08:00
__init__.py update docs 2021-01-23 15:25:58 +08:00