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

Test model weights loading

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-16 21:07:06 +02:00
parent 9fb9d7e8e1
commit 432adf57bd
2 changed files with 43 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/model/model.cpp
View File

@@ -51,12 +51,7 @@ float* Model::predict(const float* input) {
void Model::addLayer(const std::string& name, Layers::SequentialLayer* layer) {
layers.push_back(layer);
Layers::WeightedLayer* wLayer = dynamic_cast<Layers::WeightedLayer*>(layer);
if (wLayer != nullptr) {
layerMap[name] = wLayer;
}
layerMap[name] = layer;
}
Layers::SequentialLayer* Model::getLayer(const std::string& name) {