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

Fix module layer issues

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-30 13:08:13 +02:00
parent 8ac2da004c
commit 479c1119e7
4 changed files with 116 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
include/model/module.hpp
View File

@@ -17,19 +17,14 @@ class Module : public Layers::SequentialLayer {
int getInputSize();
void addLayer(const std::string& name, Layers::SequentialLayer* layer);
Layers::SequentialLayer* getLayer(const std::string& name);
const std::unordered_map<std::string, Layers::SequentialLayer*>& getLayers() const;
const std::vector<std::pair<std::string, Layers::SequentialLayer*>>& getLayers() const;
protected:
int inputSize;
int inputChannels;
std::vector<std::pair<std::string, Layers::SequentialLayer*>> layers;
int outputSize;
int outputChannels;
std::vector<std::pair<std::string, Layers::SequentialLayer*>> layers;
std::unordered_map<std::string, Layers::SequentialLayer*> layerMap;
int inputSize;
};
} // namespace CUDANet