From 36c09c8805c368182aaffa04c440e3d4ba8f7042 Mon Sep 17 00:00:00 2001 From: benedict-96 Date: Thu, 28 Nov 2024 11:01:15 +0100 Subject: [PATCH] Importing _compute_loss. --- src/GeometricMachineLearning.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/GeometricMachineLearning.jl b/src/GeometricMachineLearning.jl index d13493161..17b8605b9 100644 --- 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