chore: vanilla_model added in deel/torchlip/__init__.py

deel-ai · Feb 5, 2024 · 87f78e9 · 87f78e9
1 parent effd7e6
commit 87f78e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/deel/torchlip/__init__.py b/deel/torchlip/__init__.py
@@ -55,4 +55,5 @@
     "Sequential",
     "SpectralConv2d",
     "SpectralLinear",
+    "vanilla_model",
 ]
diff --git a/deel/torchlip/modules/__init__.py b/deel/torchlip/modules/__init__.py
@@ -64,6 +64,7 @@
 from .loss import NegKRLoss
 from .module import LipschitzModule
 from .module import Sequential
+from .module import vanilla_model
 from .pooling import ScaledAdaptiveAvgPool2d
 from .pooling import ScaledAvgPool2d
 from .pooling import ScaledL2NormPool2d