Module laplace.curvature.curvlinops
Classes
class CurvlinopsInterface (model: nn.Module, likelihood: Likelihood | str, last_layer: bool = False, subnetwork_indices: torch.LongTensor | None = None, dict_key_x: str = 'input_ids', dict_key_y: str = 'labels')-
Interface for Curvlinops backend. https://github.com/f-dangel/curvlinops
Ancestors
Subclasses
Inherited members
class CurvlinopsGGN (model: nn.Module, likelihood: Likelihood | str, last_layer: bool = False, subnetwork_indices: torch.LongTensor | None = None, dict_key_x: str = 'input_ids', dict_key_y: str = 'labels', stochastic: bool = False)-
Implementation of the
GGNInterfaceusing Curvlinops.Ancestors
Inherited members
class CurvlinopsEF (model: nn.Module, likelihood: Likelihood | str, last_layer: bool = False, subnetwork_indices: torch.LongTensor | None = None, dict_key_x: str = 'input_ids', dict_key_y: str = 'labels')-
Implementation of
EFInterfaceusing Curvlinops.Ancestors
Inherited members
class CurvlinopsHessian (model: nn.Module, likelihood: Likelihood | str, last_layer: bool = False, subnetwork_indices: torch.LongTensor | None = None, dict_key_x: str = 'input_ids', dict_key_y: str = 'labels')-
Implementation of the full Hessian using Curvlinops.
Ancestors
Inherited members