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 |
||
---|---|---|
.. | ||
backbones | ||
heads | ||
losses | ||
meta_arch | ||
__init__.py |