multi-GPU training

Help scvi-tools
1
  • 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?
2

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.