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

Implement getting layer, weights and biases

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-16 19:09:41 +02:00
parent f4ae45f867
commit 9fb9d7e8e1
7 changed files with 77 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/model/model.hpp
View File

@@ -29,6 +29,8 @@ class Model {
float* predict(const float* input);
void addLayer(const std::string& name, Layers::SequentialLayer* layer);
Layers::SequentialLayer* getLayer(const std::string& name);
void loadWeights(const std::string& path);
private:
@@ -41,7 +43,7 @@ class Model {
int outputSize;
std::vector<Layers::SequentialLayer*> layers;
std::unordered_map<std::string, Layers::WeightedLayer*> layerMap;
std::unordered_map<std::string, Layers::SequentialLayer*> layerMap;
};
} // namespace CUDANet