nip.trainers.rl_tensordict_base

A generic reinforcement learning trainer using tensordicts.

Functions

update_schedule_iterator(schedule)

Return a bool iterator which specifies on which iterations to update an agent.

Classes

TensorDictRlTrainer(hyper_params, ...)

Base class for all reinforcement learning trainers which use tensordicts.