inlcude loss derivative in init

zincware · May 29, 2024 · 0fa1a4d · 0fa1a4d
1 parent c896254
commit 0fa1a4d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/papyrus/measurements/__init__.py b/papyrus/measurements/__init__.py
@@ -31,6 +31,7 @@
     NTKMagnitudeDistribution,
     NTKSelfEntropy,
     NTKTrace,
+    LossDerivative,
 )
 
 __all__ = [
@@ -43,4 +44,5 @@
     Loss.__name__,
     Accuracy.__name__,
     NTK.__name__,
+    LossDerivative.__name__,
 ]