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.