SupervisedSolver.py 🚧

Source: LION/optimizers/SupervisedSolver.py

Warning

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

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

No docstring is available.

SupervisedSolver.mini_batch_step(self, sino, target)

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

SupervisedSolver.default_parameters() → SolverParams

No docstring is available.