imitation.scripts.ingredients.wb#

This ingredient provides Weights & Biases logging.

Functions

wandb_init(_run, wandb_name_prefix, ...)

Putting everything together to get the W&B kwargs for wandb.init().
imitation.scripts.ingredients.wb.wandb_init(_run, wandb_name_prefix, wandb_tag, wandb_kwargs, wandb_additional_info, log_dir)[source]#

Putting everything together to get the W&B kwargs for wandb.init().

Parameters

  • wandb_name_prefix (str) – User-specified prefix for wandb run name.

  • wandb_tag (Optional[str]) – User-specified tag for this run.

  • wandb_kwargs (Mapping[str, Any]) – User-specified kwargs for wandb.init().

  • wandb_additional_info (Mapping[str, Any]) – User-specific additional info to add to wandb experiment config.

  • log_dir (str) – W&B logs will be stored in directory {log_dir}/wandb/.

Raises

ModuleNotFoundError – wandb is not installed.

Return type

None