multi-GPU training

  • 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.