From 6970d1b61611c455a25322b1510f53a9c8f8dcfa Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Thu, 13 Jan 2022 16:08:18 +0000 Subject: [PATCH] - fixed deep learning crash git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@806 b431acfa-c32f-4a4a-93f1-934dc6c82436 --- source/Layer.hpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/source/Layer.hpp b/source/Layer.hpp index 76cfbee..ecfcf80 100644 --- a/source/Layer.hpp +++ b/source/Layer.hpp @@ -49,6 +49,11 @@ public: } } + void train(arma::mat const &batch, IListener *pListener=nullptr) + { + Rbm::train(trainingData(batch), pListener); + } + std::string& name() { return m_name;