Menu

Benchmark multi-model/multi-view models.

Note

This page is a reference documentation. It only explains the function signature, and not how to use it. Please refer to the gallery for the big picture.

mmbench.workflow.smcvae.to_device(data, device)[source]ΒΆ

Transfer data to device.

Parameters

data : tensor or list of tensor

the data to be transfered.

device : torch.device

the device to work on.

Returns

out : tensor or list of tensor

the transfered data.

Follow us

© 2023, mmbench developers