train_eval

Description

Routines for training and evaluation of pytorch DNNs with callbacks and calibration.

Sub-modules

base_handles

Basic handles for training and evaluation of pytorch models.

callbacks

Callback functionality and typical callbacks for training.

hessian

Exact computation of full Hessian using autodiff for calibration.

kpis

Metrics and loss measures for pytorch model training and evaluation.

train_eval_funs

Basic methods for training and evaluation of models with callback calls.