deepinv.training#

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

deepinv.Trainer

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

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.