CSAILVision / CSAILVision/semantic-segmentation-pytorch
Is there any way I can test using CPU
Open
- Dominant language
- Python
- Stars
- 5.1k
- Forks
- 1.1k
- PR merge metrics
- No merged PRs in 30d
Description
Several issues here.
After I torch.cuda.set_device(-1)
segmentation_module.cuda() and test(segmentation_module, loader_test, -1) do not perform well.
Is there any way I can test using CPU?
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.