EquivariantSolver.py 🚧

Source: LION/optimizers/EquivariantSolver.py

Warning

This file has not yet received a complete narrative and docstring audit. Its public source-level API is listed automatically below.

get_rotation_matrix(angle: float)

No docstring is available.

class EquivariantSolver(model: LIONmodel, optimizer: Optimizer, loss_fn: Callable, geometry: Geometry, verbose: bool = True, device: torch.device = ..., solver_params: SolverParams | None = None)

No docstring is available.

EquivariantSolver.rotation_group(cardinality: int)

No docstring is available.

EquivariantSolver.default_parameters() → LIONParameter

No docstring is available.

EquivariantSolver.mini_batch_step(self, sino_batch, target_batch) → torch.Tensor

No docstring is available.

EquivariantSolver.cite(cite_format='MLA')

No docstring is available.