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

Update module add layer

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-19 20:22:08 +02:00
parent 870b121c2a
commit b5fb205df8
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/model/module.cpp
View File

@@ -5,8 +5,16 @@
using namespace CUDANet; using namespace CUDANet;
void Module::addLayer(const std::string& name, Layers::SequentialLayer* layer) { void Module::addLayer(const std::string& name, Layers::SequentialLayer* layer) {
layers.push_back({ name, layer }); Module* module = dynamic_cast<Module*>(layer);
layerMap[name] = layer;
if (module != nullptr) {
layers.push_back({ name, module });
for (const auto& moduleLayer : module->getLayers()) {
layerMap[moduleLayer.first] = moduleLayer.second;
}
return;
}
} }
Layers::SequentialLayer* Module::getLayer(const std::string& name) { Layers::SequentialLayer* Module::getLayer(const std::string& name) {