fast-reid/fastreid
Xingyu Liao 1dce15efad
faster dataloader with pre-fetch and cuda stream ()
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
..
config update fastreid v1.2 readme and changelog 2021-04-06 20:09:13 +08:00
data faster dataloader with pre-fetch and cuda stream () 2021-04-12 15:03:35 +08:00
engine update fastreid v1.2 2021-04-02 21:33:13 +08:00
evaluation support classification in fastreid () 2021-03-26 20:17:39 +08:00
layers update fastreid v1.2 2021-04-02 21:33:13 +08:00
modeling faster dataloader with pre-fetch and cuda stream () 2021-04-12 15:03:35 +08:00
solver update fastreid v1.2 readme and changelog 2021-04-06 20:09:13 +08:00
utils remove apex dependency () 2021-03-23 12:12:35 +08:00
__init__.py update fastreid v1.2 readme and changelog 2021-04-06 20:09:13 +08:00