Thank you for the author's work. I have two GPUs, and during inference with your model, I noticed that only one GPU is being utilized even though I changed the device to 0, 1. It seems that this doesn't work. Could you please provide a solution to enable both GPUs to work simultaneously?
Thank you for the author's work. I have two GPUs, and during inference with your model, I noticed that only one GPU is being utilized even though I changed the device to 0, 1. It seems that this doesn't work. Could you please provide a solution to enable both GPUs to work simultaneously?