ray.tune.logger.TBXLoggerCallback#

class ray.tune.logger.TBXLoggerCallback[source]#

Bases: LoggerCallback

TensorBoardX Logger.

Note that hparams will be written only after a trial has terminated. This logger automatically flattens nested dicts to show on TensorBoard:

{“a”: {“b”: 1, “c”: 2}} -> {“a/b”: 1, “a/c”: 2}

Methods

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