draugr.torch_utilities.sessions.device_sessions.TorchCpuSession

class draugr.torch_utilities.sessions.device_sessions.TorchCpuSession(model: Optional[Module] = None, no_side_effect: bool = True)[source]

Bases: AlsoDecorator

Sets global torch devices to cpu

__init__(model: Optional[Module] = None, no_side_effect: bool = True)[source]

Methods

__init__([model, no_side_effect])