ride.unfreeze
¶
Module Contents¶
Classes¶
Abstract base-class for Ride mixins |
Functions¶
|
|
|
|
|
|
|
Attributes¶
- class ride.unfreeze.Unfreezable(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.RideMixin
Abstract base-class for Ride mixins
- static configs() ride.core.Configs [source]¶
- on_init_end(hparams, layers_to_unfreeze: Sequence[Tuple[str, torch.nn.Module]] = None, names_to_unfreeze: Sequence[str] = None, *args, **kwargs)[source]¶
- ride.unfreeze.freeze_layers_except_names(parent_module: torch.nn.Module, names_to_unfreeze: Sequence[str])[source]¶
- ride.unfreeze.get_modules_to_unfreeze(parent_module: torch.nn.Module, name_must_include='') Sequence[Tuple[str, torch.nn.Module]] [source]¶
- ride.unfreeze.unfreeze_from_end(layers: Sequence[Tuple[str, torch.nn.Module]], num_layers_from_end: int, freeze_others=False)[source]¶