CSAILVision / CSAILVision/semantic-segmentation-pytorch
RuntimeError: CUDA error: out of memory
Open
- Dominant language
- Python
- Stars
- 5.1k
- Forks
- 1.1k
- PR merge metrics
- No merged PRs in 30d
Description
I just use one image to test, but get the error below:
File "test.py", line 203, in
main(cfg, args.gpu)
File "test.py", line 129, in main
test(segmentation_module, loader_test, gpu)
File "test.py", line 79, in test
scores = scores + pred_tmp / len(cfg.DATASET.imgSizes)
RuntimeError: CUDA error: out of memory
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.