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

Change model input layer creation

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-21 00:24:49 +01:00
parent af6838e8ae
commit a9d0a0832a
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/model/model.hpp
View File

@@ -19,6 +19,8 @@ class Model {
private:
Layers::Input *inputLayer;
int inputSize;
int inputChannels;

9
src/model/model.cpp
View File

@@ -12,19 +12,20 @@ Model::Model(const int inputSize, const int inputChannels)
const int inputLayerSize = inputSize * inputSize * inputChannels;
Layers::Input* inputLayer = new Layers::Input(inputLayerSize);
layers.push_back(inputLayer);
inputLayer = new Layers::Input(inputLayerSize);
};
Model::~Model(){};
float* Model::predict(const float* input) {
float* d_input = inputLayer->forward(input);
for (auto& layer : layers) {
input = layer->forward(input);
d_input = layer->forward(d_input);
}
return d_input;
}
void Model::addLayer(const std::string& name, Layers::SequentialLayer* layer) {