search

Kiteretsu77 / APISR

APISR: Anime Production Inspired Real-World Anime Super-Resolution (CVPR 2024)
GNU General Public License v3.0
748 stars 51 forks source link

Multi GPU support ? #14

Open hekmon opened 3 weeks ago

hekmon commented 3 weeks ago

The DAT model can be very heavy, even on a 3090, when a lots of images needs to be upscalled. Is there any chance you could implements multi-gpu in order for a second card to be active ?

I have no clue how to use torch multi-gpu myself.

Thanks.

Kiteretsu77 commented 1 week ago

I used pytorch lightning to implement the multi-gpu before, instead of torch multi-gpu you said. The challenge would be lied on the sync for the degradation model. However, I have to say that I also only use one 24 GB memory GPU to train the DAT model. You need to decrease the batch size such that it is trainable.