geowatch.utils.lightning_ext.callbacks.text_logger module¶

class geowatch.utils.lightning_ext.callbacks.text_logger.TextLogger(args=None)[source]¶

Bases: Callback

Writes logging information to text files.

CommandLine

xdoctest -m geowatch.utils.lightning_ext.callbacks.text_logger TextLogger

Example

>>> #
>>> from geowatch.utils.lightning_ext.callbacks.text_logger import *  # NOQA
>>> from geowatch.utils.lightning_ext import demo
>>> from geowatch.monkey import monkey_lightning
>>> monkey_lightning.disable_lightning_hardware_warnings()
>>> self = demo.LightningToyNet2d(num_train=55)
>>> default_root_dir = ub.Path.appdir('lightning_ext/tests/TextLogger').ensuredir()
>>> #
>>> trainer = pl.Trainer(callbacks=[TextLogger()],
>>>                      default_root_dir=default_root_dir,
>>>                      max_epochs=3, accelerator='cpu', devices=1)
>>> trainer.fit(self)
>>> text_logs = ub.Path(trainer.text_logger.log_fpath).read_text()
>>> print(text_logs)

setup(trainer: Trainer, pl_module: LightningModule, stage: str | None = None) → None[source]¶

teardown(trainer: Trainer, pl_module: LightningModule, stage: str | None = None) → None[source]¶

on_fit_start(trainer: Trainer, pl_module: LightningModule) → None[source]¶

on_fit_end(trainer: Trainer, pl_module: LightningModule) → None[source]¶

state_dict()[source]¶

load_state_dict(checkpoint)[source]¶

on_train_start(trainer: Trainer, pl_module: LightningModule) → None[source]¶

on_train_end(trainer: Trainer, pl_module: LightningModule) → None[source]¶

on_sanity_check_start(trainer: Trainer, pl_module: LightningModule) → None[source]¶

on_sanity_check_end(trainer: Trainer, pl_module: LightningModule) → None[source]¶

on_exception(trainer: Trainer, pl_module: LightningModule, *args, **kw) → None[source]¶

on_train_epoch_start(trainer: Trainer, pl_module: LightningModule) → None[source]¶

on_train_epoch_end(trainer: Trainer, pl_module: LightningModule) → None[source]¶

on_validation_epoch_end(trainer: Trainer, pl_module: LightningModule) → None[source]¶

on_validation_epoch_start(trainer: Trainer, pl_module: LightningModule) → None[source]¶