StableSR icon indicating copy to clipboard operation
StableSR copied to clipboard

Increasing Batch Size using Multiple GPUs

Open JorgeMarmol opened this issue 1 year ago • 1 comments

Hello, I would like to use a larger batch size (currently set at 5), but I encounter an "out of memory" issue when I have more than 1 GPU available. Does anyone know how I can utilize all my GPUs in the process to increase the batch size? @IceClear

JorgeMarmol avatar Nov 14 '23 09:11 JorgeMarmol

Currently the code does not support multi-GPU test. You may refer to the training script to modify it to support multi-GPU test or you can run the same script multiple times on different GPUs.

IceClear avatar Nov 28 '23 16:11 IceClear