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

PadConv2D #52

Closed cofri closed 2 years ago

cofri commented 2 years ago

This PR introduces a new PadConv2D layer that is the combination of a padding operation and a Conv2D layer. Different padding types can be set:

Moreover, some bugs are fixed and unit tests are added.

thib-s commented 2 years ago

As PadConv2D is not lipschitz constrained, we need to find a way to make sure that users are aware of it. Should we move it to a separate module for instance ?