mypackage.MyPyroModel.to_device¶
-
MyPyroModel.
to_device
(device)¶ Move model to device.
- Parameters
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