command
python tools/test_ins.py configs/solov2/solov2_light_448_r34_fpn_8gpu_3x.py work_dirs/solov2_light_release_r34_fpn_8gpu_3x/epoch_36.pth --show --out results_solo.pkl
--eval segm
bug
[>>>>>>>>>>>>> ] 20/76, 0.3 task/s, elapsed: 59s, ETA: 165s
Traceback (most recent call last):
...
RuntimeError: CUDA out of memory. Tried to allocate 3.30 GiB (GPU 0; 8.00 GiB total capacity; 973.14 MiB already allocated; 2.13 GiB free; 3.74 GiB reserved in total by PyTorch)
Then I shrinked my test set to 14 images, same error occurred when [>> ] 2/14.
Environment
python 3.7
CUDA 11.1
PyTorch 1.7.0+cu110
Supplement
The epoch_36.pth file is generated from the training on my own dataset. And performed pretty good when single-tested by inference_demo.py but fail with this batch-test command.