Testing with Multi-GPU

[YongLD]

I meet an OOM error when I make a test with python demo/restoration_video_demo.py.

RuntimeError: CUDA out of memory. Tried to allocate 32.04 GiB (GPU 0; 6.00 GiB total capacity; 1.64 GiB already allocated; 3.37 GiB free; 1.65 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

So I would like to make a test with multi-GPU, but I just do not know how to change the script with ./tools/dist_test.sh ${CONFIG_FILE} ${CHECKPOINT_FILE} ${GPU_NUM} [--out ${RESULT_FILE}] [--save-path ${IMAGE_SAVE_PATH}]

How can I make a restoration test with multi-gpu?

[wangruohui]

test is mainly used to compute evaluation metrics and probably not solving your OOM problem. Did you use this demo on long video or large resolution ones?

[zengyh1900]

Closing due to inactivity, please reopen if there are any further problems.