scvelo.tl.VELOVI.to_device¶
-
VELOVI.
to_device
(device)¶ Move model to device.
Examples
>>> adata = scvi.data.synthetic_iid()
>>> model = scvi.model.SCVI(adata)
>>> model.to_device('cpu') # moves model to CPU
>>> model.to_device('cuda:0') # moves model to GPU 0
>>> model.to_device(0) # also moves model to GPU 0