wandb_osh.hooks

Module Contents

Classes

TriggerWandbSyncHook

Hook to trigger synchronization of wandb with wandb-osh

class wandb_osh.hooks.TriggerWandbSyncHook(communication_dir: os.PathLike = _comm_default_dir)

Hook to trigger synchronization of wandb with wandb-osh

Parameters:

communication_dir – Directory used for communication with wandb-osh.

__call__(logdir: str | os.PathLike | None = None)

Trigger synchronization on the head nodes

Parameters:

logdir – The directory in which wandb puts its run files.

Returns:

None