Auxiliary Modules

Auxiliary Modules#

The following modules provide support to the main library, or implement additional, secondary functionality.

nip.rl_objectives

Implementations of RL objectives, extending those of TorchRL.

nip.utils

Various utility functions and classes.

nip.stat_logger

Classes for logging experiment statistics during training.

nip.artifact_logger

Logging training artifacts to W&B every so often.

nip.base_run

Utilities for basing the current experiment on a previous W&B run.

nip.model_cache

Utilities for saving and loading models together with their parameters.

nip.timing

Code for timing various things, to test performance.