new layers torchlip level

deel-ai · Jun 17, 2024 · d292e9d · d292e9d
1 parent b4191a6
commit d292e9d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/deel/torchlip/__init__.py b/deel/torchlip/__init__.py
@@ -37,9 +37,12 @@
     "FullSort",
     "GroupSort",
     "GroupSort2",
+    "BatchCentering",
+    "LayerCentering",
     "HKRLoss",
     "HKRMulticlassLoss",
     "HingeMarginLoss",
+    "HKRMultiLoss",
     "HingeMulticlassLoss",
     "InvertibleDownSampling",
     "InvertibleUpSampling",
@@ -52,7 +55,9 @@
     "ScaledAdaptiveAvgPool2d",
     "ScaledAvgPool2d",
     "ScaledL2NormPooling2D",
+    "ScaledGlobalL2NormPooling2D",
     "Sequential",
     "SpectralConv2d",
+    "SpectralConv1d",
     "SpectralLinear",
 ]