Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement model destructor

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-14 00:05:57 +02:00
parent 53c976733b
commit b20ade27d8
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/model/model.cpp
View File

@@ -25,7 +25,13 @@ Model::Model(const Model& other)
outputLayer = new Layers::Output(*other.outputLayer); 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* Model::predict(const float* input) {
float* d_input = inputLayer->forward(input); float* d_input = inputLayer->forward(input);