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

Fix/minor fixes #54

Closed thib-s closed 2 years ago

thib-s commented 2 years ago

some quick fixes:

  1. added unconstrained layers to init.py for correct imports
  2. added missing rst files for metrics and unconstrained layers
  3. added atol to losses test in order to ensure consistent tests