mirror of https://github.com/JDAI-CV/fast-reid.git
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 |
||
---|---|---|
.. | ||
__init__.py | ||
baseline.py | ||
build.py | ||
distiller.py | ||
mgn.py | ||
moco.py |