draugr.torch_utilities.sessions.device_sessions

Created on 11/05/2020

Classes

TorchCpuSession([model, no_side_effect])

Sets global torch devices to cpu

TorchCudaSession([model, no_side_effect])

Sets global torch devices to cuda if available

TorchDeviceSession(device[, model, ...])

Sets global torch devices to cpu