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 |
||
---|---|---|
.. | ||
config | ||
data | ||
engine | ||
evaluation | ||
layers | ||
modeling | ||
solver | ||
utils | ||
__init__.py |