- Does any of the provided method support multi-GPU training？
- I am trying to develop my custom model with your fantastic tools. I noticed that in scvi.trainer, it has an option to set gpus
, how should I use this feature to build a model supporting multi-GPU training?
So ideally you’d just set
gpus>1 but our data loader does not currently work for multi-gpu training. It’s something we are currently looking into.