Increase number of CPU cores for vae.train() in scvi?

Help scvi-tools
,
1

When I run vae.train(use_gpu=False), how can I increase number of CPU cores for parallel processing?

2

Hi, could you clarify what you mean by v_num?

3

Thank you for replying. I meant how can I increase the number of CPU cores when I train the model?

4

Hi, you can specify this with

vae.train(accelerator="cpu", devices=num_cpus)

This feature is not available in a release yet but on the main branch, so you would have to install from source.