StableSR
StableSR copied to clipboard
Increasing Batch Size using Multiple GPUs
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
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.