mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 17:54:27 +00:00
Fix adding module layers
This commit is contained in:
@@ -56,17 +56,18 @@ float* Model::predict(const float* input) {
|
||||
}
|
||||
|
||||
void Model::addLayer(const std::string& name, Layers::SequentialLayer* layer) {
|
||||
Module* module = dynamic_cast<Module*>(layer);
|
||||
const Module* module = dynamic_cast<Module*>(layer);
|
||||
|
||||
if (module != nullptr) {
|
||||
layers.push_back({name, module});
|
||||
for (const auto& moduleLayer : module->getLayers()) {
|
||||
layerMap[moduleLayer.first] = moduleLayer.second;
|
||||
layers.push_back({moduleLayer.first, moduleLayer.second});
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
layers.push_back({name, layer});
|
||||
layerMap[name] = layer;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user