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

Refactor vanilla export for model #49

Closed cofri closed 2 years ago

cofri commented 2 years ago

The vanilla export functions for models are refactored and simplified by using the tf.keras.model.clone_model() function. Unit tests are added to verify that model outputs from a Lipschitz model and its vanilla equivalent are identical.