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

..

__init__.py

update fastreid V1.0

2021-01-18 11:36:38 +08:00

baseline.py

faster dataloader with pre-fetch and cuda stream (#456)

2021-04-12 15:03:35 +08:00

build.py

add configurable decorator & linear loss decouple (#441)

2021-03-23 12:10:06 +08:00

distiller.py

add configurable decorator & linear loss decouple (#441)

2021-03-23 12:10:06 +08:00

mgn.py

faster dataloader with pre-fetch and cuda stream (#456)

2021-04-12 15:03:35 +08:00

moco.py

add configurable decorator & linear loss decouple (#441)

2021-03-23 12:10:06 +08:00