SelfSupervisedSolver.py 🚧

Source: LION/optimizers/SelfSupervisedSolver.py

Warning

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

class SelfSupervisedSolver(model: LIONmodel, optimizer: Optimizer, loss_fn, geometry: Geometry = None, verbose: bool = False, model_regularization=None, device: torch.device = None)

No docstring is available.

SelfSupervisedSolver.mini_batch_step(self, sino, target)

This function is responsible for performing a single mini-batch step of the optimization. returns the loss of the mini-batch

SelfSupervisedSolver.default_parameters() → SolverParams

No docstring is available.