to_empty

UNet.to_empty(*, device: Union[str, torch.device]) → torch.nn.modules.module.T

Moves the parameters and buffers to the specified device without copying storage.

Parameters

device (torch.device) – The desired device of the parameters and buffers in this module.

Returns

self

Return type

Module