lightning_template.utils.callbacks.set_sharing_strategy_callback

Classes

SetSharingStrategyCallback

Abstract base class used to build new callbacks.

Module Contents

class lightning_template.utils.callbacks.set_sharing_strategy_callback.SetSharingStrategyCallback(strategy: str | None = 'file_descriptor')

Bases: lightning.pytorch.Callback

Abstract base class used to build new callbacks.

Subclass this class and override any of the relevant hooks

strategy = 'file_descriptor'