wandb_osh.lightning_hooks

Module Contents

Classes

TriggerWandbSyncLightningCallback

Hook to be used when interfacing wandb with Lightning.

class wandb_osh.lightning_hooks.TriggerWandbSyncLightningCallback(communication_dir: os.PathLike = _comm_default_dir)

Bases: lightning.pytorch.Callback

Hook to be used when interfacing wandb with Lightning.

Parameters:

communication_dir – Directory used for communication with wandb-osh.

Usage

from wandb_osh.lightning_hooks import TriggerWandbSyncLightningCallback

trainer = Trainer(callbacks=[TriggerWandbSyncLightningCallback()])

on_validation_epoch_end(trainer: lightning.pytorch.Trainer, pl_module: lightning.pytorch.LightningModule) → None