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

Add layer name to vector

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-21 12:20:02 +02:00
parent 0170afaf3f
commit 942ee6a32b
2 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
include/model/model.hpp
View File

@@ -11,7 +11,10 @@
namespace CUDANet {
enum TensorType { WEIGHT, BIAS, };
enum TensorType {
WEIGHT,
BIAS,
};
struct TensorInfo {
std::string name;
@@ -42,7 +45,7 @@ class Model {
int outputSize;
std::vector<Layers::SequentialLayer*> layers;
std::vector<std::pair<std::string, Layers::SequentialLayer*>> layers;
std::unordered_map<std::string, Layers::SequentialLayer*> layerMap;
};

14
src/model/model.cpp
View File

@@ -15,7 +15,7 @@ Model::Model(const int inputSize, const int inputChannels, const int outputSize)
: inputSize(inputSize),
inputChannels(inputChannels),
outputSize(outputSize),
layers(std::vector<Layers::SequentialLayer*>()),
layers(std::vector<std::pair<std::string, Layers::SequentialLayer*>>()),
layerMap(std::unordered_map<std::string, Layers::SequentialLayer*>()) {
inputLayer = new Layers::Input(inputSize * inputSize * inputChannels);
outputLayer = new Layers::Output(outputSize);
@@ -25,7 +25,7 @@ Model::Model(const Model& other)
: inputSize(other.inputSize),
inputChannels(other.inputChannels),
outputSize(other.outputSize),
layers(std::vector<Layers::SequentialLayer*>()),
layers(std::vector<std::pair<std::string, Layers::SequentialLayer*>>()),
layerMap(std::unordered_map<std::string, Layers::SequentialLayer*>()) {
inputLayer = new Layers::Input(*other.inputLayer);
outputLayer = new Layers::Output(*other.outputLayer);
@@ -34,8 +34,8 @@ Model::Model(const Model& other)
Model::~Model() {
delete inputLayer;
delete outputLayer;
for (auto layer : layers) {
delete layer;
for (const auto& layer : layers) {
delete layer.second;
}
};
@@ -43,15 +43,15 @@ float* Model::predict(const float* input) {
float* d_input = inputLayer->forward(input);
for (auto& layer : layers) {
std::cout << layer << std::endl;
d_input = layer->forward(d_input);
std::cout << layer.first << std::endl;
d_input = layer.second->forward(d_input);
}
return outputLayer->forward(d_input);
}
void Model::addLayer(const std::string& name, Layers::SequentialLayer* layer) {
layers.push_back(layer);
layers.push_back({ name, layer });
layerMap[name] = layer;
}