CSAILVision / CSAILVision/semantic-segmentation-pytorch
test gpu to cpu(.cpu()）is very slow
Open
- Dominant language
- Python
- Stars
- 5.1k
- Forks
- 1.1k
- PR merge metrics
- No merged PRs in 30d
Description
[https://github.com/CSAILVision/semantic-segmentation-pytorch/blob/master/test.py](url)
82 line
pred = as_numpy(pred.squeeze(0).cpu())
.cpu()is very slow
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.