Batch-wise inference (multiple images in parallel) support

[YuQiao0303]

Hi authors, thank you for this amazing work!
I notice that in 'main_ddpir_inpainting.py', there is a for loop that deal with one image at a time. If we have a lot of images to deal with, this may be relatively slow. I wonder if it's possible to process multiple images as a minibatch in parallel? Thank you!!

[yuanzhi-zhu]

Hi @YuQiao0303 ,

Thanks for your interest, could you please try e.g. python main_ddpir.py --opt configs/inpaint.yaml in the latest updated repository? You can try to change the batch size in the config file inpaint.yaml :)