base_handles

Description

Basic handles for training and evaluation of pytorch models. These are wrappers around the training and evaluation functions from train_eval_funs, with additional support of automated several epoch training.

Sub-modules

early_stopping

Handle for early stopping.

resettable_optimizer

Wrapper around pytorch optimizer to make them resettable.

train_test_handle

Abstract handle for standard training and testing of pytorch models.