cyto_dl.callbacks.layer_freeze module#

class cyto_dl.callbacks.layer_freeze.LayerFreeze(modules: str | List[str] = [], contains: str | List[str] | None = None, excludes: str | List[str] | None = None)[source]#

Bases: Callback

Parameters:

  • modules (Union[str, List[str]]) – List of modules to search within

  • contains (Optional[Union[str, List[str]]]) – List of strings that must be contained in the layer name to freeze

  • excludes (Optional[Union[str, List[str]]]) – List of strings that must not be contained in the layer name to freeze

setup(trainer, pl_module, stage)[source]#