nip.utils.checkpoints

Utilities for dealing with saved experiment checkpoints.

Currently these utilities only work with checkpoints produced by pure-text trainers, where rollouts are stored in a NestedArrayDict.

Functions

download_checkpoint(wandb_project, run_id[, ...])	Download a checkpoint from wandb to the local checkpoint directory.
load_rollouts()	Load the rollouts from a checkpoint.
load_run_hyper_parameters(run_id, wandb_project)	Load the hyper-parameters of a run from W&B, converting them as needed.