lightning_template.utils.callbacks.set_sharing_strategy_callback

Module Contents

Classes

SetSharingStrategyCallback

Set sharing strategy when training starts.

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

Bases: lightning.pytorch.Callback

Set sharing strategy when training starts.