-
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
use cuda version torch.randperm to avoid copy from gpu to cpu and a fatal bug in multi-thread cpu version
| Name |
Last commit
|
Last update |
|---|---|---|
| .. | ||
| config | Loading commit data... | |
| csrc | Loading commit data... | |
| data | Loading commit data... | |
| engine | Loading commit data... | |
| layers | Loading commit data... | |
| modeling | Loading commit data... | |
| solver | Loading commit data... | |
| structures | Loading commit data... | |
| utils | Loading commit data... | |
| __init__.py | Loading commit data... |