LayerSync

class lightning.pytorch.plugins.LayerSync

기반 클래스: abc.ABC

Abstract base class for creating plugins that wrap layers of a model with synchronization logic for multiprocessing.

abstract apply(model)

Override this method to apply synchronization to the layers of this model.

반환 형식

Module

abstract revert(model)

Override this method to undo all modifications made in apply().

반환 형식

Module