lit_sb3.callbacks.summary_writer

source module lit_sb3.callbacks.summary_writer

Callback exposing TensorBoard SummaryWriter from Lightning TensorBoardLogger.

Classes

• SummaryWriterCallback — Resolve and expose the active TensorBoard SummaryWriter.

source class SummaryWriterCallback()

Bases : L.Callback

Resolve and expose the active TensorBoard SummaryWriter.

Attributes

• state_key : str — Identifier for the state of the callback.

Methods

• on_train_start

• on_validate_start

• on_test_start

• on_predict_start

• get_writer

source method SummaryWriterCallback.on_train_start(trainer: L.Trainer, pl_module: L.LightningModule) → None

source method SummaryWriterCallback.on_validate_start(trainer: L.Trainer, pl_module: L.LightningModule) → None

source method SummaryWriterCallback.on_test_start(trainer: L.Trainer, pl_module: L.LightningModule) → None

source method SummaryWriterCallback.on_predict_start(trainer: L.Trainer, pl_module: L.LightningModule) → None

source classmethod SummaryWriterCallback.get_writer(trainer: L.Trainer) → Any