geowatch.utils.lightning_ext.callbacks.state_logger module

class geowatch.utils.lightning_ext.callbacks.state_logger.StateLogger

Bases: Callback

Prints out what callbacks are being called

DEPRECATE: Use text_logger

setup(trainer: Trainer, pl_module: LightningModule, stage: str | None = None) → None

teardown(trainer: Trainer, pl_module: LightningModule, stage: str | None = None) → None

on_fit_start(trainer: Trainer, pl_module: LightningModule) → None

on_fit_end(trainer: Trainer, pl_module: LightningModule) → None

on_save_checkpoint(trainer: Trainer, pl_module: LightningModule, checkpoint: Dict[str, Any]) → dict

on_sanity_check_start(trainer: Trainer, pl_module: LightningModule) → None

on_sanity_check_end(trainer: Trainer, pl_module: LightningModule) → None

on_exception(trainer: Trainer, pl_module: LightningModule, *args, **kw) → None