From b20ade27d88500af5dfab0071099aa82264999c8 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Sun, 14 Apr 2024 00:05:57 +0200
Subject: [PATCH] Implement model destructor

---
 src/model/model.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/model/model.cpp b/src/model/model.cpp
index 46fd799..cd02236 100644
--- a/src/model/model.cpp
+++ b/src/model/model.cpp
@@ -25,7 +25,13 @@ Model::Model(const Model& other)
     outputLayer = new Layers::Output(*other.outputLayer);
 }
 
-Model::~Model(){};
+Model::~Model(){
+    delete inputLayer;
+    delete outputLayer;
+    for (auto layer : layers) {
+        delete layer;
+    }
+};
 
 float* Model::predict(const float* input) {
     float* d_input = inputLayer->forward(input);