Multi-GPU support

[aksloms]

Hi,

Thank you so much for sharing your great work!

I would like to ask if you are planning to add multi-gpu support for training? I keep running out of memory for the default settings, so I think this would be a good option :)

Thanks again!
Joanna

[ocetintas]

Hi,

This is a good point and we might look into this in the future. However, I cannot promise anything about the timeline.

In the meantime, you can reduce the batch size or use SUSHI with 7 levels and 128 frames. This setting requires less than 16GB of GPU memory. The performance will be just slightly lower but you can still reach SOTA results.