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
GGNInterface
using 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
EFInterface
using 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