deepinv.unfolded#

This module provides networks architectures based on unfolding optimization algorithms. Please refer to user guide for more details.

Unfolded#

User Guide: refer to Unfolded Algorithms for more information.

deepinv.unfolded.unfolded_builder

Helper function for building an unfolded architecture.

deepinv.unfolded.BaseUnfold

Base class for unfolded algorithms.

Deep Equilibrium#

User Guide: refer to Deep Equilibrium for more information.

deepinv.unfolded.DEQ_builder

Helper function for building an instance of the BaseDEQ() class.

deepinv.unfolded.BaseDEQ

Base class for deep equilibrium (DEQ) algorithms.

Custom Unfolded Blocks#

User Guide: refer to Custom Unfolded Architectures for more information.

deepinv.models.PDNet_PrimalBlock

Primal block for the Primal-Dual unfolding model.

deepinv.models.PDNet_DualBlock

Dual block for the Primal-Dual unfolding model.