-
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... |