nip.trainers.trainer_base

nip.trainers.trainer_base#

Base classes for all trainers.

Functions

attach_progress_bar(num_iterations_func)

Decorate a Trainer method to attach a progress bar.

Classes

IterationContext(trainer, num_iterations)

Context manager for methods that run for a certain number of iterations.

TensorDictTrainer(hyper_params, ...)

Base class for trainers that use tensors and TensorDicts.

Trainer(hyper_params, scenario_instance, ...)

Base class for all trainers.

Exceptions

CheckPointNotFoundError

Exception raised when a checkpoint is not found.