draugr.torch_utilities.sessions.model_sessions

Created on 11/05/2020

Classes

TorchEvalSession(model[, no_side_effect])

# speed up evaluating after training finished

TorchTrainSession(model[, no_side_effect])

# speed up evaluating after training finished

TorchFrozenModelSession(model[, no_side_effect])

description

TorchTrainingSession

alias of TorchTrainSession