-
Jiayuan Gu authored
use cuda version torch.randperm to avoid copy from gpu to cpu and a fatal bug in multi-thread cpu version
b4182125
| Name |
Last commit
|
Last update |
|---|---|---|
| .. | ||
| backbone | ||
| detector | ||
| roi_heads | ||
| rpn | ||
| __init__.py | ||
| balanced_positive_negative_sampler.py | ||
| box_coder.py | ||
| matcher.py | ||
| poolers.py | ||
| utils.py |