Importing _compute_loss.

JuliaGNI · Nov 28, 2024 · 36c09c8 · 36c09c8
1 parent d0abe33
commit 36c09c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/GeometricMachineLearning.jl b/src/GeometricMachineLearning.jl
@@ -29,7 +29,7 @@ module GeometricMachineLearning
     import AbstractNeuralNetworks: parameterlength
     import AbstractNeuralNetworks: GlorotUniform
     import AbstractNeuralNetworks: params, architecture, model, dim
-    import AbstractNeuralNetworks: AbstractPullback, NetworkLoss
+    import AbstractNeuralNetworks: AbstractPullback, NetworkLoss, _compute_loss
     # export params, architetcure, model
     export dim
     import GeometricIntegrators.Integrators: method, GeometricIntegrator