search

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

Do not warn users if Keras 1-Lipschitz layers are used #55

Closed cofri closed 2 years ago

cofri commented 2 years ago

Some Keras layers are 1-Lipschitz, e.g. ReLU, MaxPool2D or Flatten. These layers can be safely used in a Lipschitz network. When creating a deel.lip.Sequential model, a warning message is raised for not Lipschitz layers. This warning is not raised anymore for supported 1-Lipschitz Keras layers.