Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +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: private:
Layers::Input *inputLayer;
int inputSize; int inputSize;
int inputChannels; 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; const int inputLayerSize = inputSize * inputSize * inputChannels;
Layers::Input* inputLayer = new Layers::Input(inputLayerSize); inputLayer = new Layers::Input(inputLayerSize);
layers.push_back(inputLayer);
}; };
Model::~Model(){}; Model::~Model(){};
float* Model::predict(const float* input) { float* Model::predict(const float* input) {
float* d_input = inputLayer->forward(input);
for (auto& layer : layers) { 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) { void Model::addLayer(const std::string& name, Layers::SequentialLayer* layer) {