Bering.models.ImageEncoder.to_empty
- ImageEncoder.to_empty(*, device, recurse=True)
Moves the parameters and buffers to the specified device without copying storage.
- Return type:
TypeVar(T, bound= Module)
- Args:
- device (
torch.device): The desired device of the parameters and buffers in this module.
- recurse (bool): Whether parameters and buffers of submodules should
be recursively moved to the specified device.
- device (
- Returns:
Module: self