DiffBIR icon indicating copy to clipboard operation
DiffBIR copied to clipboard

SR scale x1 with OOM: CUDA out of memory

Open amanoooo opened this issue 2 years ago • 4 comments

I want to handle my photo(resolution 4080 × 3060), using RTX 4090, enabled Tiled, it failed!

What should i do? Does DiffBIR support quantization ?

amanoooo avatar Oct 24 '23 05:10 amanoooo

Hey, try tiled options for inference. It can avoid memory limitation.

ziyannchen avatar Nov 08 '23 08:11 ziyannchen

Hey, try tiled options for inference. It can avoid memory limitation.

hello.what is that mean?

learn01one avatar Mar 22 '24 07:03 learn01one

Hey, try tiled options for inference. It can avoid memory limitation.

hello.what is that mean?

If you wanna inference large-resolution images with limited memory, try to specify --tiled related params as follows.

python inference.py \
.... \
--tiled --tile_size 512 --tile_stride 256

ziyannchen avatar Mar 22 '24 10:03 ziyannchen

@ziyannchen 请问 tiled sample 是否意味着对图像分片超分?例如一张图像是1024x1024,tile_stride=256,那么就会把图像分成四份,对每份进行超分,然后拼接?

jiaxiangc avatar Apr 07 '24 03:04 jiaxiangc