draugr.torch_utilities.sessions.jit_sessions.TorchJitSession

class draugr.torch_utilities.sessions.jit_sessions.TorchJitSession(enabled=False, no_side_effect: bool = True)[source]

Bases: AlsoDecorator

# Disable torch jit tracing

__init__(enabled=False, no_side_effect: bool = True)[source]

Methods

__init__([enabled, no_side_effect])