draugr.torch_utilities.sessions.cache_sessions.TorchCacheSession

class draugr.torch_utilities.sessions.cache_sessions.TorchCacheSession(using_cuda: bool = True)[source]

Bases: AlsoDecorator

# speed up evaluating after training finished # NOTE: HAS THE SIDE EFFECT OF CLEARING CACHE, NON RECOVERABLE

__init__(using_cuda: bool = True)[source]

Methods

__init__([using_cuda])