ray.tune.logger.aim.AimLoggerCallback#

class ray.tune.logger.aim.AimLoggerCallback(repo: str | None = None, experiment_name: str | None = None, metrics: List[str] | None = None, **aim_run_kwargs)[source]#

Bases: LoggerCallback

Aim Logger: logs metrics in Aim format.

Aim is an open-source, self-hosted ML experiment tracking tool. It’s good at tracking lots (thousands) of training runs, and it allows you to compare them with a performant and well-designed UI.

Source: aimhubio/aim

Methods

`__init__`	Initialize the Aim logger callback.
`get_state`	Get the state of the callback.
`log_trial_restore`	Handle logging when a trial restores.
`log_trial_save`	Handle logging when a trial saves a checkpoint.
`on_checkpoint`	Called after a trial saved a checkpoint with Tune.
`on_experiment_end`	Called after experiment is over and all trials have concluded.
`on_step_begin`	Called at the start of each tuning loop step.
`on_step_end`	Called at the end of each tuning loop step.
`on_trial_recover`	Called after a trial instance failed (errored) but the trial is scheduled for retry.
`set_state`	Set the state of the callback.
`setup`	Called once at the very beginning of training.

Attributes

`VALID_HPARAMS`
`VALID_NP_HPARAMS`