ray.tune.logger.TBXLoggerCallback#

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

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`