deel-ai / deel-lip

Build and train Lipschitz constrained networks: TensorFlow implementation of k-Lipschitz layers
https://deel-ai.github.io/deel-lip/
MIT License
89 stars 10 forks source link

SpectralConv2DTranspose #59

Closed cofri closed 2 years ago

cofri commented 2 years ago

The SpectralConv2DTranspose layer is a new Lipschitz layer equivalent to the tf.keras Conv2DTranspose. Note that some arguments must be set to a specific value to be supported:

Like the SpectralConv2D layer, the kernel is normalized using the spectral normalization, then orthogonalized with Björck algorithm.

thib-s commented 2 years ago

rebased branch onto develop