deepinv.training

This module contains the training and testing functions. Please refer to the user guide for more information.

deepinv.Trainer

Trainer(model, physics, optimizer, train_dataloader, ...) Trainer class for training a reconstruction network.

deepinv.train	Alias function for training a model using deepinv.Trainer class.
deepinv.test	Tests a reconstruction model (algorithm or network).

Adversarial Training

deepinv.training.AdversarialTrainer	AdversarialTrainer(model, physics, optimizer, train_dataloader, losses_d, D, step_ratio_D, ...) Trainer class for training a reconstruction network using adversarial learning.
deepinv.training.AdversarialOptimizer	Optimizer for adversarial training that encapsulates both generator and discriminator's optimizers.