From 87f78e9392b00b10d595ff385eeb65c91885b155 Mon Sep 17 00:00:00 2001 From: Yannick Prudent Date: Mon, 5 Feb 2024 17:22:28 +0100 Subject: [PATCH] chore: vanilla_model added in deel/torchlip/__init__.py --- deel/torchlip/__init__.py | 1 + deel/torchlip/modules/__init__.py | 1 + 2 files changed, 2 insertions(+) diff --git a/deel/torchlip/__init__.py b/deel/torchlip/__init__.py index be2c5b0..72c0e52 100644 --- 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 index b2653d3..7f730d4 100644 --- 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